ei.ddlZddlmZmZmZddlmZ ddlmZddl m Z ddl mZmZmZmZmZmZmZmZddlmZej0eZeZej9d ed g eefd ed efdZej=ded g eddeddeddedddeddddeefdededededed ef d Z ej=d!ed g eefd"e!d efd#Z"y#e $r ddl mZYwxYw)$N) APIRouterDependsQuery)Session)get_db) controller) ChatRequest ChatResponseChatDataChatErrorResponseChatHistoryListResponseChatHistoryDetailResponseChatSessionListItemChatSessionDetail)chat_history_servicez/chatzSmart Inventory - Frontend)response_modeltagsrequestdbc   tj||j|j|j|j |j |j}tdtdi|dS#t$r}tjdt|ddlm}|j}|. |j!|j}n#t$rd }YnwxYwtd t|dd t| d cYd}~Sd}~wwxYw)a^ AI-powered chat endpoint for querying inventory data using natural language. Supports continuous conversation with context memory. All conversations are persisted to the database. **Session Management:** - **First message**: Send `chat_session_id: null` to start a new conversation - **Response**: You'll receive a `chat_session_id` in the response - **Follow-up**: Send the same `chat_session_id` to continue the conversation - **New chat**: Send `chat_session_id: null` again to start fresh - **Context**: Last 5 questions and answers are sent to LLM for context **Example questions:** - "What are the top 10 fast-moving products?" - "Tell me more about the first one" (follow-up) - "What about location 2?" (follow-up with context) - "Show me products that are out of stock" **Request body:** ```json { "chat_session_id": null, // or existing session ID "company_id": 1, "user_id": 1, "store_id": 1, "branch_id": 1, "question": "What are my top selling products?" } ``` )r company_idquestionuser_idstore_id branch_idchat_session_idTzQuery processed successfullysuccessdatamessagez"Error in inventory chat endpoint: r)chat_session_serviceNz error-sessionFz2An error occurred while processing your question: )rraw_dataanswerzError processing query)r process_chat_queryrrrrrrr r Exceptionloggererrorstrservicesr"create_session)rrresulter" session_ids G/var/www/html/hubwallet-dev/src/smart_inventory/apps/chat_bot/router.pyinventory_chatr1sF% ..))%%OO%%''#33 #F#2    9#a&BC 3,,   -1@@ASAST  -,  - *KCPQF8T -   sBA.A11 D :5D0C  D CDC(DD D z /chat/history.zUser ID) descriptionzStore IDz Branch IDzPage number (1-indexed))ger2dzNumber of items per page)r3ler2rrrpage page_sizec B tj||||||\}}td|Dcgc] }td i|c}|||dScc}w#t$rH} t j dt| tdgd||dt| cYd } ~ Sd } ~ wwxYw) a  Get paginated list of chat sessions for a user/store/branch. Returns chat sessions ordered by most recently updated first. Each session includes: - `chat_session_id`: Use this to open/continue a chat - `chat_name`: Derived from the first question asked - `message_count`: Total number of messages in the conversation **Usage:** - Display as a list of previous chats in the sidebar - Click on a chat to load its details using the `/chat/history/{chat_session_id}` endpoint )rrrrr7r8Tz#Chat history retrieved successfully)rr totalr7r8r!z!Error getting chat history list: FrzError retrieving chat history: Nr%)rget_chat_history_listrrr'r(r)r*) rrrr7r8rsessionsr:sr.s r0r;r;bs, .DD  %'4<=q%**=9   =    8QAB&5c!fX>     s-(A A A A B=BBBz/chat/history/{chat_session_id}rc " tj||}|tdddStdtd i|dS#t$rE}t j dt|tddd t|cYd}~Sd}~wwxYw) aZ Get detailed chat session with all messages. Use this endpoint when user clicks on a chat from the history list. Returns all messages in chronological order. **Usage:** - Load all messages to display the full conversation - User can continue the chat by sending new messages with the same `chat_session_id` )rrNFzChat session not foundrTz#Chat session retrieved successfullyz#Error getting chat history detail: zError retrieving chat session: r%)rget_chat_session_detailrrr'r(r)r*)rrsession_detailr.s r0get_chat_history_detailrAs -EE+   !,0  )"4^49    :3q6(CD(5c!fX>   s"&AA B :B B B)#loggingfastapirrrsqlalchemy.ormr src.utils.dbr ImportErrorutils.dbr schemasr r r r rrrrr+r getLogger__name__r(routerpostr1getintr;r*rAr%r0rQs--" # +   8 $ W\9U8VW&/G G G XG TO,CKgJhi)4#:63K8aA+DE2!9ST&/ 1 1 1 1   1  1  1 j1 h ->W_{^|}&/' ' ' ~' c  sD DD