Update app.py

app.py

@@ -644,35 +644,97 @@ class AdaptiveMedicalBot:
             logger.error(f"Error updating learning system: {e}")
 
 def create_demo():
-    """Create Gradio interface"""
-    bot = AdaptiveMedicalBot()
-    
-    def chat(message, history):
-        try:
-            response_data = bot.generate_response(message, history)
-            return response_data['response']
-        except Exception as e:
-            logger.error(f"Chat error: {e}")
-            return "I apologize, but I'm experiencing technical difficulties. For emergencies, please call 999."
-
-    def handle_feedback(feedback: str, history: List[tuple]):
-        if history:
-            last_interaction = history[-1]
-            bot.update_from_feedback(last_interaction[0], last_interaction[1], 1 if feedback == "👍" else -1)
+    """Create Gradio interface with proper initialization"""
+    try:
+        bot = AdaptiveMedicalBot()
+        
+        def chat(message, history, feedback=None):
+            try:
+                # Handle feedback if provided
+                if feedback and history:
+                    last_interaction = history[-1]
+                    bot.update_from_feedback(
+                        last_interaction[0],
+                        last_interaction[1],
+                        1 if feedback == "👍" else -1
+                    )
+                
+                # For first message, provide introduction
+                if not history:
+                    return "Hello! I'm Pearly, your NHS medical assistant. How can I help you today?"
+                
+                response = bot.generate_response(message, history)
+                
+                # Clean up response formatting
+                response = response.strip()
+                if response.startswith("Response:"):
+                    response = response[9:].strip()
+                    
+                return response
+                
+            except Exception as e:
+                logger.error(f"Chat error: {e}")
+                return "I apologize, but I'm experiencing technical difficulties. For emergencies, please call 999."
 
-    demo = gr.ChatInterface(
-        fn=chat,
-        title="NHS Medical Triage Assistant",
-        description="Hello! I'm Pearly, I'm here to help assess your symptoms and provide appropriate healthcare guidance. Please describe your concerns.",
+        # Create Gradio interface with better styling
+        demo = gr.Interface(
+            fn=chat,
+            inputs=[
+                gr.Textbox(
+                    label="Your message",
+                    placeholder="Type your message here...",
+                    lines=2
+                ),
+                gr.State([]),  # For chat history
+                gr.Radio(
+                    choices=["👍", "👎"],
+                    label="Was this response helpful?",
+                    visible=True,
+                    value=None
+                )
+            ],
+            outputs=gr.Chatbot(
+                label="NHS Medical Triage Assistant",
+                height=500
+            ),
+            title="NHS Medical Triage Assistant - Pearly",
+            description="""
+            👋 Hello! I'm Pearly, your NHS medical assistant.
+            
+            I can help you with:
+            • Assessing your symptoms
+            • Finding appropriate care
+            • Booking appointments
+            • General medical guidance
+            
+            For emergencies, always call 999.
+            For urgent concerns, call NHS 111.
+            
+            Please describe your concerns below.
+            """,
+            examples=[
+                ["I've been having headaches recently"],
+                ["I need to book a routine checkup"],
+                ["I'm feeling very anxious lately"]
+            ],
+            theme="soft"
+        )
         
-        additional_inputs=[
-            gr.Radio(["👍", "👎"], label="Was this response helpful?", visible=True)
-        ],
-        additional_inputs_update_fun=handle_feedback
-    )
-    
-    return demo
+        return demo
+        
+    except Exception as e:
+        logger.error(f"Error creating demo: {e}")
+        raise
 
 if __name__ == "__main__":
+    # Initialize environment
+    load_dotenv()
+    
+    # Set up HuggingFace login if token exists
+    hf_token = os.getenv("HF_TOKEN")
+    if hf_token:
+        login(token=hf_token)
+    
+    # Launch demo
     demo = create_demo()
     demo.launch()