Update components/chat.py

components/chat.py

@@ -6,6 +6,7 @@ from utils.database import verify_vector_store
 from threading import Lock
 from typing import Optional
 import traceback
+from utils.response_formatter import ResponseFormatter, display_formatted_response
 
 # Create a lock for QA system access
 qa_lock = Lock()
@@ -37,6 +38,31 @@ def _verify_chat_ready() -> bool:
         return False
     return True
 
+def _process_user_message(prompt: str):
+    """Process a new user message and generate AI response."""
+    try:
+        with st.spinner("Analyzing..."):
+            human_message = HumanMessage(content=prompt)
+            st.session_state.messages.append(human_message)
+            with st.chat_message("user"):
+                st.write(prompt)
+
+            with qa_lock:
+                response = st.session_state.qa_system.invoke({
+                    "input": prompt,
+                    "chat_history": st.session_state.messages
+                })
+
+            if response:
+                # Format the response
+                ai_message = AIMessage(content=str(response))
+                st.session_state.messages.append(ai_message)
+                with st.chat_message("assistant"):
+                    display_formatted_response(
+                        str(response),
+                        metadata=getattr(response, 'metadata', None)
+                    )
+
 def _display_chat_history():
     """Display all messages in the chat history."""
     for message in st.session_state.messages: