Update app.py

app.py

@@ -1,7 +1,6 @@
 import os
 import sys
 import signal
-import asyncio
 import warnings
 import gradio as gr
 import config
@@ -108,6 +107,13 @@ footer { visibility: hidden !important; }
     fill: currentColor;
 }
 
+/* Chatbot styling */
+#chatbot {
+    min-height: 500px;
+    max-height: 500px;
+    overflow-y: auto;
+}
+
 /* Mobile responsiveness */
 @media (max-width: 768px) {
     .dev-footer div {
@@ -118,6 +124,11 @@ footer { visibility: hidden !important; }
     .dev-footer a {
         margin: 5px 0;
     }
+    
+    #chatbot {
+        min-height: 400px;
+        max-height: 400px;
+    }
 }
 """
 
@@ -160,8 +171,6 @@ def create_ui():
         
         # Get response
         try:
-            # Note: chat_response doesn't typically check history in current logic, 
-            # but if it did, we'd need to ensure it handles the new format or pass just strings.
             response = chat_response(message, history)
         except Exception as e:
             response = f"Error: {str(e)}"
@@ -185,8 +194,34 @@ def create_ui():
             
         return history, ""
 
+    def regenerate_response(history):
+        """Regenerate the last assistant response."""
+        if not history or len(history) < 2:
+            return history
+        
+        # Get the last user message
+        last_user_msg = None
+        for msg in reversed(history):
+            if msg["role"] == "user":
+                last_user_msg = msg["content"]
+                break
+        
+        if last_user_msg:
+            # Remove the last assistant response
+            if history[-1]["role"] == "assistant":
+                history.pop()
+            
+            # Generate new response
+            try:
+                response = chat_response(last_user_msg, history)
+                history.append({"role": "assistant", "content": response})
+            except Exception as e:
+                history.append({"role": "assistant", "content": f"Error: {str(e)}"})
+        
+        return history
+
     # Create the Gradio Blocks
-    with gr.Blocks(title="⚖️ LegalizeAI", css=custom_css) as demo:
+    with gr.Blocks(title="⚖️ LegalizeAI") as demo:
         gr.Markdown(
             """
             # ⚖️ LegalizeAI
@@ -201,8 +236,8 @@ def create_ui():
         chatbot = gr.Chatbot(
             height=500,
             elem_id="chatbot",
-            avatar_images=(None, "⚖️"),
-            show_copy_button=True
+            avatar_images=(None, "⚖️")
+            # Removed show_copy_button for compatibility with older Gradio versions
         )
         
         with gr.Row():
@@ -221,32 +256,6 @@ def create_ui():
             clear_btn = gr.Button("Clear 🗑️", size="sm")
             regenerate_btn = gr.Button("Regenerate 🔁", size="sm")
 
-        def regenerate_response(history):
-            """Regenerate the last assistant response."""
-            if not history or len(history) < 2:
-                return history
-            
-            # Get the last user message
-            last_user_msg = None
-            for msg in reversed(history):
-                if msg["role"] == "user":
-                    last_user_msg = msg["content"]
-                    break
-            
-            if last_user_msg:
-                # Remove the last assistant response
-                if history[-1]["role"] == "assistant":
-                    history.pop()
-                
-                # Generate new response
-                try:
-                    response = chat_response(last_user_msg, history)
-                    history.append({"role": "assistant", "content": response})
-                except Exception as e:
-                    history.append({"role": "assistant", "content": f"Error: {str(e)}"})
-            
-            return history
-
         # Example buttons logic
         examples = [
             "What is the punishment for theft in Pakistan?",
@@ -318,11 +327,12 @@ def main():
             server_name=server_name,
             server_port=7860,
             share=False,  # Don't use share=True on Spaces
-            favicon_path="⚖️" if is_huggingface else None,
+            favicon_path=None,
             theme=theme,
             show_error=True,
             quiet=False,
-            show_api=False
+            show_api=False,
+            css=custom_css  # CSS moved here for Gradio 6.0+ compatibility
         )
         
     except KeyboardInterrupt: