Change chatbot

app.py

@@ -518,11 +518,15 @@ def create_interface():
                 with gr.Row():
                     with gr.Column(scale=3):
                         gr.Markdown("**Note**: You are chatting with the currently loaded model. If you've just completed processing, you're testing the modified model. To test the original model, reload it in the Model Processing tab.")
-                        chatbot = gr.Chatbot(
-                            label="Chat Window",
-                            height=400,
-                            type="messages"
+                        
+                        # Use Textbox instead of Chatbot for better compatibility
+                        chat_display = gr.Textbox(
+                            label="Chat History",
+                            lines=20,
+                            interactive=False,
+                            value="Chat history will appear here..."
                         )
+                        
                         msg = gr.Textbox(
                             label="Input Message",
                             placeholder="Enter your question...",
@@ -577,30 +581,42 @@ def create_interface():
             outputs=[process_output, process_image]
         )
         
-        # Chat functionality with streaming
-        def user(user_message, history):
-            return "", history + [{"role": "user", "content": user_message}]
+        # Chat functionality with simple text display
+        def user(user_message, chat_history):
+            if chat_history == "Chat history will appear here...":
+                chat_history = ""
+            new_history = chat_history + f"\n\n👤 User: {user_message}"
+            return "", new_history
         
-        def bot(history, max_new_tokens, temperature):
-            if history and history[-1]["role"] == "user":
-                # Get complete response first
-                response, _ = processor.chat(history[-1]["content"], history[:-1], max_new_tokens, temperature)
+        def bot(chat_history, max_new_tokens, temperature):
+            # Extract the last user message
+            lines = chat_history.split('\n')
+            user_message = None
+            for line in reversed(lines):
+                if line.startswith('👤 User: '):
+                    user_message = line[9:]  # Remove "👤 User: " prefix
+                    break
+            
+            if user_message:
+                # Get complete response
+                response, _ = processor.chat(user_message, [], max_new_tokens, temperature)
                 print(f"DEBUG: Bot function received response: {response[:200]}...")
                 print(f"DEBUG: Bot function full response: {response}")
-                print(f"DEBUG: Bot function history before append: {history}")
-                history.append({"role": "assistant", "content": response})
-                print(f"DEBUG: Bot function history after append: {history}")
-            return history
+                
+                # Add assistant response to chat history
+                new_history = chat_history + f"\n\n🤖 Assistant: {response}"
+                return new_history
+            return chat_history
         
-        msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-            bot, [chatbot, max_new_tokens, temperature], chatbot
+        msg.submit(user, [msg, chat_display], [msg, chat_display], queue=False).then(
+            bot, [chat_display, max_new_tokens, temperature], chat_display
         )
         
-        send_btn.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-            bot, [chatbot, max_new_tokens, temperature], chatbot
+        send_btn.click(user, [msg, chat_display], [msg, chat_display], queue=False).then(
+            bot, [chat_display, max_new_tokens, temperature], chat_display
         )
         
-        clear.click(lambda: [], None, chatbot, queue=False)
+        clear.click(lambda: "Chat history will appear here...", None, chat_display, queue=False)
         
         # Bind organization selection event
         export_to_org.change(