Critical fix: Remove gr.Chatbot() to avoid Gradio 4.x schema bug - use Textbox instead

app.py

@@ -1,6 +1,7 @@
 """
 Personality Chatbot - Multi-personality LLM with LoRA adapters
 Deployed on Hugging Face Spaces
+FIXED: Removed gr.Chatbot() to avoid Gradio 4.x schema bug
 """
 
 import gradio as gr
@@ -81,10 +82,10 @@ def switch_personality(personality_name):
     except Exception as e:
         return f"❌ Error loading adapter: {str(e)}"
 
-def generate_response(message, history, temperature=0.7, max_tokens=256):
+def generate_response(message, chat_history, temperature=0.7, max_tokens=256):
     """Generate response using current personality"""
     if current_adapter is None:
-        return "⚠️ Please select a personality first!"
+        return "⚠️ Please select a personality first!", chat_history
     
     try:
         # Format with chat template
@@ -96,12 +97,12 @@ def generate_response(message, history, temperature=0.7, max_tokens=256):
         with torch.no_grad():
             outputs = current_adapter.generate(
                 **inputs,
-                max_new_tokens=max_tokens,  # Changed from max_length to max_new_tokens
+                max_new_tokens=max_tokens,
                 temperature=temperature,
                 do_sample=True,
                 top_p=0.9,
-                top_k=50,  # Added top-k sampling
-                repetition_penalty=1.1,  # Added to reduce repetition
+                top_k=50,
+                repetition_penalty=1.1,
                 pad_token_id=tokenizer.pad_token_id,
                 eos_token_id=tokenizer.eos_token_id,
             )
@@ -114,10 +115,14 @@ def generate_response(message, history, temperature=0.7, max_tokens=256):
         response = response.replace("<|im_start|>", "").replace("<|im_end|>", "")
         response = response.replace("assistant\n", "").strip()
         
-        return response
+        # Format chat history for display
+        new_history = chat_history + f"\n\n**You:** {message}\n\n**Bot ({current_personality}):** {response}\n\n---"
+        
+        return response, new_history
     
     except Exception as e:
-        return f"❌ Error generating response: {str(e)}"
+        error_msg = f"❌ Error generating response: {str(e)}"
+        return error_msg, chat_history
 
 def handle_personality_change(personality_name):
     """Handle personality dropdown change"""
@@ -128,7 +133,7 @@ def handle_personality_change(personality_name):
 print("🚀 Starting application...")
 load_base_model()
 
-# Create Gradio interface
+# Create Gradio interface WITHOUT gr.Chatbot() to avoid schema bug
 with gr.Blocks(theme=gr.themes.Soft(), title="Personality Chatbot") as demo:
     gr.Markdown(
         """
@@ -182,9 +187,13 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Personality Chatbot") as demo:
             )
         
         with gr.Column(scale=2):
-            chatbot = gr.Chatbot(
-                label="Chat",
-                height=500,
+            # FIXED: Use Textbox instead of Chatbot to avoid schema bug
+            chat_display = gr.Textbox(
+                label="💬 Chat History",
+                value="",
+                lines=20,
+                max_lines=30,
+                interactive=False,
                 show_label=True,
             )
             
@@ -194,6 +203,14 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Personality Chatbot") as demo:
                 lines=2,
             )
             
+            last_response = gr.Textbox(
+                label="🤖 Last Response",
+                value="",
+                lines=5,
+                interactive=False,
+                show_label=True,
+            )
+            
             with gr.Row():
                 submit_btn = gr.Button("Send 💬", variant="primary")
                 clear_btn = gr.Button("Clear 🗑️", variant="secondary")
@@ -201,11 +218,10 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Personality Chatbot") as demo:
     # Event handlers
     def respond(message, chat_history, temperature, max_tokens):
         if not message.strip():
-            return chat_history, ""
+            return "", chat_history, ""
         
-        bot_response = generate_response(message, chat_history, temperature, max_tokens)
-        chat_history.append((message, bot_response))
-        return chat_history, ""
+        bot_response, new_history = generate_response(message, chat_history, temperature, max_tokens)
+        return "", new_history, bot_response
     
     # Personality change handler
     personality_dropdown.change(
@@ -217,21 +233,21 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Personality Chatbot") as demo:
     # Chat handlers
     submit_btn.click(
         fn=respond,
-        inputs=[msg_box, chatbot, temperature_slider, max_tokens_slider],
-        outputs=[chatbot, msg_box],
+        inputs=[msg_box, chat_display, temperature_slider, max_tokens_slider],
+        outputs=[msg_box, chat_display, last_response],
     )
     
     msg_box.submit(
         fn=respond,
-        inputs=[msg_box, chatbot, temperature_slider, max_tokens_slider],
-        outputs=[chatbot, msg_box],
+        inputs=[msg_box, chat_display, temperature_slider, max_tokens_slider],
+        outputs=[msg_box, chat_display, last_response],
     )
     
     clear_btn.click(
-        fn=lambda: ([], ""),
-        outputs=[chatbot, msg_box],
+        fn=lambda: ("", "", ""),
+        outputs=[chat_display, msg_box, last_response],
     )
 
-# Launch
+# Launch with share=True for HF Spaces
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860, share=False)