Update app.py

app.py

@@ -163,8 +163,8 @@ def retrieve_relevant_chunks(query, top_k=3):
         print(f"Error retrieving chunks: {e}")
         return [], []
 
-def respond(message, history):
-    """Chat function - returns messages format"""
+def chat(message, history):
+    """Chat function - processes message and returns response string"""
     global client
     
     # Reinitialize client if needed
@@ -178,17 +178,17 @@ def respond(message, history):
             pass
     
     if client is None:
-        return {"role": "assistant", "content": "[ERROR] Groq API not initialized. Set GROQ_API_KEY in Settings."}
+        return "[ERROR] Groq API not initialized. Set GROQ_API_KEY in Settings."
     
     if not document_store['chunks']:
-        return {"role": "assistant", "content": "[WARNING] Please upload and process documents first."}
+        return "[WARNING] Please upload and process documents first."
     
     try:
         # Retrieve context
         relevant_chunks, metadata = retrieve_relevant_chunks(message, top_k=3)
         
         if not relevant_chunks:
-            return {"role": "assistant", "content": "[ERROR] No relevant information found."}
+            return "[ERROR] No relevant information found."
         
         # Build context
         context = "\n\n".join([
@@ -201,11 +201,10 @@ def respond(message, history):
             {"role": "system", "content": "You are a helpful assistant that answers questions based on provided context. Be concise and accurate."}
         ]
         
-        # Add history
+        # Add history - history is already in messages format
         if history:
             for msg in history[-6:]:  # Last 6 messages
-                if isinstance(msg, dict) and "role" in msg and "content" in msg:
-                    messages.append({"role": msg["role"], "content": msg["content"]})
+                messages.append(msg)
         
         # Add current query
         messages.append({
@@ -237,11 +236,12 @@ def respond(message, history):
             'answer': answer
         })
         
-        return {"role": "assistant", "content": full_answer}
+        # Return just the string - ChatInterface handles adding to history
+        return full_answer
         
     except Exception as e:
         print(f"Error: {e}")
-        return {"role": "assistant", "content": f"[ERROR] {str(e)}"}
+        return f"[ERROR] {str(e)}"
 
 def download_history():
     """Download chat history"""
@@ -255,7 +255,7 @@ def download_history():
     except:
         return None
 
-# Build interface
+# Build interface - USE gr.ChatInterface!
 with gr.Blocks(title="Enhanced RAG Chatbot") as demo:
     
     gr.Markdown("""
@@ -278,28 +278,23 @@ with gr.Blocks(title="Enhanced RAG Chatbot") as demo:
             gr.Markdown("### History")
             download_btn = gr.Button("Download (JSON)")
             download_file = gr.File(label="Download")
-            clear_btn = gr.Button("Clear Chat")
         
         with gr.Column(scale=2):
-            chatbot = gr.Chatbot(label="Conversation", height=500)
-            msg = gr.Textbox(label="Ask a question", lines=2)
-            submit = gr.Button("Ask", variant="primary")
+            # Use ChatInterface directly - it handles everything!
+            chat_interface = gr.ChatInterface(
+                fn=chat,
+                type="messages",
+                chatbot=gr.Chatbot(height=500, label="Conversation"),
+                textbox=gr.Textbox(label="Ask a question", placeholder="Type your question here...", lines=2),
+                submit_btn="Ask",
+                retry_btn=None,
+                undo_btn=None,
+                clear_btn="Clear Chat"
+            )
     
     # Process files
     process_btn.click(process_files, [file_upload], [process_output])
     
-    # Chat - using respond that returns dict format
-    submit.click(respond, [msg, chatbot], [chatbot]).then(
-        lambda: "", None, [msg]
-    )
-    
-    msg.submit(respond, [msg, chatbot], [chatbot]).then(
-        lambda: "", None, [msg]
-    )
-    
-    # Clear
-    clear_btn.click(lambda: [], None, chatbot)
-    
     # Download
     download_btn.click(download_history, None, [download_file])