Update app.py

app.py

@@ -641,11 +641,11 @@ class AdaptiveMedicalBot:
             logger.error(f"Error updating learning system: {e}")
 
 def create_demo():
-    """Create Gradio interface with proper message formatting"""
+    """Create enhanced Gradio interface with advanced features and styling"""
     try:
         bot = AdaptiveMedicalBot()
         
-        def chat(message: str, history: List[Dict[str, str]]):
+        def chat(message: str, history: List[Dict[str, str]], urgency_filter: str = "all"):
             try:
                 # Convert history to the format expected by the bot
                 bot_history = [(h["user"], h["bot"]) for h in history] if history else []
@@ -667,8 +667,8 @@ def create_demo():
                     {"role": "assistant", "content": "I apologize, but I'm experiencing technical difficulties. For emergencies, please call 999."}
                 ]
 
-        def process_feedback(feedback: str, history: List[Dict[str, str]]):
-            """Process feedback when received"""
+        def process_feedback(feedback: str, history: List[Dict[str, str]], comment: str = ""):
+            """Process feedback with optional comment"""
             try:
                 if history and len(history) >= 2:
                     last_user_msg = history[-2]["content"]
@@ -676,98 +676,222 @@ def create_demo():
                     bot.update_from_feedback(
                         last_user_msg,
                         last_bot_msg,
-                        1 if feedback == "👍" else -1
+                        1 if feedback == "👍" else -1,
+                        comment=comment
                     )
             except Exception as e:
                 logger.error(f"Error processing feedback: {e}")
 
-        # Create Gradio interface with proper message handling
-        with gr.Blocks(theme="soft") as demo:
-            gr.Markdown("""
-            # GP Medical Triage Assistant - Pearly
-            
-            👋 Hello! I'm Pearly, your GP 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 111.
-            
-            Please describe your concerns below.
+        # Create enhanced Gradio interface
+        with gr.Blocks(theme=gr.themes.Soft(
+            primary_hue="blue",
+            secondary_hue="indigo",
+            neutral_hue="slate",
+            font=gr.themes.GoogleFont("Inter")
+        )) as demo:
+            # Custom CSS for enhanced styling
+            gr.HTML("""
+                <style>
+                    .container { max-width: 900px; margin: auto; }
+                    .header { text-align: center; padding: 20px; }
+                    .emergency-banner {
+                        background-color: #ff4444;
+                        color: white;
+                        padding: 10px;
+                        text-align: center;
+                        font-weight: bold;
+                        margin-bottom: 20px;
+                    }
+                    .features-grid {
+                        display: grid;
+                        grid-template-columns: repeat(auto-fit, minmax(200px, 1fr));
+                        gap: 20px;
+                        padding: 20px;
+                    }
+                    .feature-card {
+                        background: #f8f9fa;
+                        padding: 15px;
+                        border-radius: 10px;
+                        text-align: center;
+                    }
+                </style>
             """)
             
-            # Initialize chat components
-            chatbot = gr.Chatbot(
-                value=[{"role": "assistant", "content": "Hello! I'm Pearly, your GP medical assistant. How can I help you today?"}],
-                height=500,
-                type="messages"
-            )
-            msg = gr.Textbox(
-                label="Your message",
-                placeholder="Type your message here...",
-                lines=2
-            )
+            # Emergency Banner
+            gr.HTML("""
+                <div class="emergency-banner">
+                    🚨 For medical emergencies, always call 999 immediately 🚨
+                </div>
+            """)
+
+            # Header Section
+            with gr.Row(elem_classes="header"):
+                gr.Markdown("""
+                    # GP Medical Triage Assistant - Pearly
+                    
+                    Welcome to your personal medical triage assistant. I'm here to help assess your symptoms and guide you to appropriate care.
+                """)
+
+            # Main Features Grid
+            gr.HTML("""
+                <div class="features-grid">
+                    <div class="feature-card">
+                        🏥 GP Appointments
+                    </div>
+                    <div class="feature-card">
+                        🔍 Symptom Assessment
+                    </div>
+                    <div class="feature-card">
+                        ⚡ Urgent Care Guide
+                    </div>
+                    <div class="feature-card">
+                        💊 Medical Advice
+                    </div>
+                </div>
+            """)
+
+            # Chat Interface
             with gr.Row():
-                submit = gr.Button("Submit")
-                clear = gr.Button("Clear")
-                
-            feedback = gr.Radio(
-                choices=["👍", "👎"],
-                label="Was this response helpful?",
-                visible=True
-            )
-            
-            # Set up event handlers
+                with gr.Column(scale=4):
+                    chatbot = gr.Chatbot(
+                        value=[{
+                            "role": "assistant",
+                            "content": "Hello! I'm Pearly, your GP medical assistant. How can I help you today?"
+                        }],
+                        height=500,
+                        elem_id="chatbot",
+                        type="messages",
+                        show_label=False
+                    )
+                    
+                    with gr.Row():
+                        msg = gr.Textbox(
+                            label="Your message",
+                            placeholder="Type your message here...",
+                            lines=2,
+                            scale=4
+                        )
+                        submit = gr.Button("Send", variant="primary", scale=1)
+
+                with gr.Column(scale=1):
+                    # Quick Actions Panel
+                    gr.Markdown("### Quick Actions")
+                    emergency_btn = gr.Button("🚨 Emergency Info", variant="secondary")
+                    nhs_111_btn = gr.Button("📞 NHS 111 Info", variant="secondary")
+                    booking_btn = gr.Button("📅 GP Booking", variant="secondary")
+                    
+                    # Conversation Controls
+                    gr.Markdown("### Controls")
+                    clear = gr.Button("🗑️ Clear Chat")
+                    
+                    # Feedback Section
+                    gr.Markdown("### Feedback")
+                    feedback = gr.Radio(
+                        choices=["👍", "👎"],
+                        label="Was this response helpful?",
+                        visible=True
+                    )
+                    feedback_text = gr.Textbox(
+                        label="Additional comments (optional)",
+                        placeholder="Tell us more about your experience...",
+                        lines=2
+                    )
+
+            # Examples Section
+            with gr.Accordion("Example Messages", open=False):
+                gr.Examples(
+                    examples=[
+                        ["I've been having severe headaches for the past week"],
+                        ["I need to book a routine checkup"],
+                        ["I'm feeling very anxious lately and need help"],
+                        ["My child has had a fever for 2 days"],
+                        ["I need information about COVID-19 testing"]
+                    ],
+                    inputs=msg
+                )
+
+            # Information Accordions
+            with gr.Accordion("NHS Services Guide", open=False):
+                gr.Markdown("""
+                    ### Emergency Services (999)
+                    - Life-threatening emergencies
+                    - Severe injuries
+                    - Suspected heart attack or stroke
+                    
+                    ### NHS 111
+                    - Urgent but non-emergency situations
+                    - Medical advice needed
+                    - Unsure where to go
+                    
+                    ### GP Services
+                    - Routine check-ups
+                    - Non-urgent medical issues
+                    - Prescription renewals
+                """)
+
+            # Event Handlers
             submit.click(
                 chat,
                 inputs=[msg, chatbot],
-                outputs=[chatbot],
-                queue=False
+                outputs=[chatbot]
             ).then(
                 lambda: "",
                 None,
-                msg  # Clear input box after sending
+                msg
             )
             
             msg.submit(
                 chat,
                 inputs=[msg, chatbot],
-                outputs=[chatbot],
-                queue=False
+                outputs=[chatbot]
             ).then(
                 lambda: "",
                 None,
-                msg  # Clear input box after sending
+                msg
             )
             
             clear.click(
                 lambda: [[], ""],
                 None,
-                [chatbot, msg],
-                queue=False
+                [chatbot, msg]
             )
             
             feedback.change(
                 process_feedback,
-                inputs=[feedback, chatbot],
-                outputs=[],
-                queue=False
+                inputs=[feedback, chatbot, feedback_text],
+                outputs=[]
             )
             
-            # Add example messages
-            gr.Examples(
-                examples=[
-                    "I've been having headaches recently",
-                    "I need to book a routine checkup",
-                    "I'm feeling very anxious lately"
-                ],
-                inputs=msg
-            )
-        
+            # Quick Action Button Handlers
+            def show_emergency_info():
+                return """🚨 Emergency Services (999)
+                - For life-threatening emergencies
+                - Severe chest pain
+                - Difficulty breathing
+                - Severe bleeding
+                - Loss of consciousness
+                """
+            
+            def show_nhs_111_info():
+                return """📞 NHS 111 Service
+                - Available 24/7
+                - Medical advice
+                - Local service information
+                - Urgent care guidance
+                """
+            
+            def show_booking_info():
+                return """📅 GP Booking Options
+                - Online booking
+                - Phone booking
+                - Routine appointments
+                - Urgent appointments
+                """
+            
+            emergency_btn.click(lambda: show_emergency_info(), outputs=[msg])
+            nhs_111_btn.click(lambda: show_nhs_111_info(), outputs=[msg])
+            booking_btn.click(lambda: show_booking_info(), outputs=[msg])
+
         return demo
         
     except Exception as e: