fixed gr message

quick_check.py

@@ -0,0 +1,16 @@
+import gradio as gr
+
+print("Gradio Version:", gr.__version__)
+
+import gradio as gr
+
+history = [
+    gr.ChatMessage(role="assistant", content="How can I help you?"),
+    gr.ChatMessage(role="user", content="Can you make me a plot of quarterly sales?"),
+    gr.ChatMessage(role="assistant", content="I am happy to provide you that report and plot.")
+]
+
+with gr.Blocks() as demo:
+    gr.Chatbot(history)
+
+demo.launch()

ui/gradio_components.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from langchain_core.messages import HumanMessage, AIMessage
 from core.rag_agent import RAGAgent
+import traceback
 
 # Initialize components
 rag_agent = None
@@ -20,12 +21,14 @@ def chat_with_agent(message, history):
     try:
         agent = initialize_agent()
         
-        # Convert Gradio history format to LangChain messages
+        # Convert Gradio history format (dict with role/content) to LangChain messages
         messages = []
-        for user_msg, assistant_msg in history:
-            messages.append(HumanMessage(content=user_msg))
-            if assistant_msg:
-                messages.append(AIMessage(content=assistant_msg))
+        if history:
+            for msg_dict in history:
+                if msg_dict["role"] == "user":
+                    messages.append(HumanMessage(content=msg_dict["content"]))
+                elif msg_dict["role"] == "assistant":
+                    messages.append(AIMessage(content=msg_dict["content"]))
         
         # Add current user message
         messages.append(HumanMessage(content=message))
@@ -53,20 +56,28 @@ def chat_with_agent(message, history):
             rag_method = result.get("rag_method", "UNKNOWN")
             response_with_metadata = f"{response}\n\n*[Source: {rag_method}]*"
             
-            # Return history in Gradio's format [[user, bot], [user, bot], ...]
-            new_history = history + [[message, response_with_metadata]]
+            # Return history in Gradio's dict format
+            new_history = history + [
+                {"role": "user", "content": message},
+                {"role": "assistant", "content": response_with_metadata}
+            ]
             return new_history
         else:
-            new_history = history + [[message, "⚠️ No response generated. Please try again."]]
+            new_history = history + [
+                {"role": "user", "content": message},
+                {"role": "assistant", "content": "⚠️ No response generated. Please try again."}
+            ]
             return new_history
             
     except Exception as e:
         error_msg = f"❌ Error: {str(e)}"
         print(f"Chat error: {e}")
-        import traceback
         traceback.print_exc()
         
-        new_history = history + [[message, error_msg]]
+        new_history = history + [
+            {"role": "user", "content": message},
+            {"role": "assistant", "content": error_msg}
+        ]
         return new_history
 
 def reset_conversation():
@@ -134,7 +145,7 @@ def create_gradio_ui():
         
         # Event handlers
         def submit_message(message, history):
-            """Handle message submission with status update"""
+            """Handle message submission"""
             if not message.strip():
                 return history, ""