Update app.py

app.py

@@ -36,7 +36,8 @@ class AdaptiveMedicalBot:
         self.setup_models()
         self.load_datasets()
         self.setup_adaptive_learning()
-        self.conversation_history = []  # Maintain conversation history
+        self.conversation_history = []  # Store conversation history
+        self.symptom_tracker = {}  # Track symptoms and severity across conversation
 
     class AdaptiveBotConfig:
         MODEL_NAME = "google/gemma-7b"
@@ -199,7 +200,7 @@ class AdaptiveMedicalBot:
             logger.error(f"Error processing feedback: {e}")
 
 def create_demo():
-    """Set up Gradio interface for the chatbot"""
+    """Set up Gradio interface for the chatbot with enhanced styling and functionality."""
     try:
         bot = AdaptiveMedicalBot()
 
@@ -216,47 +217,173 @@ def create_demo():
                 logger.error(f"Chat error: {e}")
                 return history + [
                     {"role": "user", "content": message},
-                    {"role": "assistant", "content": "I'm experiencing technical difficulties. For emergencies, call 999."}
+                    {"role": "assistant", "content": "I'm experiencing technical difficulties. For emergencies, please call 999."}
                 ]
 
         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"]
                     last_bot_msg = history[-1]["content"]
-                    bot.handle_feedback(last_user_msg, last_bot_msg, 1 if feedback == "👍" else -1)
+                    bot.handle_feedback(
+                        last_user_msg,
+                        last_bot_msg,
+                        1 if feedback == "👍" else -1
+                    )
             except Exception as e:
                 logger.error(f"Error processing feedback: {e}")
 
-        with gr.Blocks() as demo:
-            chatbot = gr.Chatbot(value=[{"role": "assistant", "content": "Hello! I'm Pearly, your GP Triage medical assistant. How can I help you today?"}],
-                height=500,
-                elem_id="chatbot",
-                type="messages",
-                show_label=False
-            )
-            
-            msg = gr.Textbox(
-                label="Your message",
-                placeholder="Type your message here...",
-                lines=2
-            )
-            submit = gr.Button("Send", variant="primary")
-            
-            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
-            )
-            
+        # 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>
+            """)
+
+            # 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():
+                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(
-                fn=chat,
+                chat,
                 inputs=[msg, chatbot],
                 outputs=[chatbot]
             ).then(
@@ -266,7 +393,7 @@ def create_demo():
             )
             
             msg.submit(
-                fn=chat,
+                chat,
                 inputs=[msg, chatbot],
                 outputs=[chatbot]
             ).then(
@@ -275,46 +402,47 @@ def create_demo():
                 msg
             )
             
+            clear.click(
+                lambda: [[], ""],
+                None,
+                [chatbot, msg]
+            )
+            
             feedback.change(
-                fn=process_feedback,
+                process_feedback,
                 inputs=[feedback, chatbot, feedback_text],
                 outputs=[]
             )
-
-            # Clear Chat Handler
-            clear = gr.Button("🗑️ Clear Chat")
-            clear.click(lambda: [[], ""], None, [chatbot, msg])
-
-            # Additional Information Sections
-            gr.HTML("""
-                <div style="padding: 20px;">
-                    <h2>Quick Actions</h2>
-                    <button onclick="document.getElementById('chatbot').value += 'Emergency Info: For emergencies, call 999.'">Emergency Info</button>
-                    <button onclick="document.getElementById('chatbot').value += 'NHS 111 Info: For urgent but non-emergency situations, call 111.'">NHS 111 Info</button>
-                    <button onclick="document.getElementById('chatbot').value += 'GP Booking Info: For routine appointments with your GP.'">GP Booking</button>
-                </div>
-            """)
-
-            gr.Markdown("### Example Messages")
-            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
-            )
-
-            gr.Markdown("""
-                ### NHS Services Guide
-                **999 - Emergency Services**: For life-threatening emergencies, severe injuries, heart attack, stroke.
-                
-                **NHS 111**: Available 24/7 for urgent but non-life-threatening situations, medical advice, and guidance.
-                
-                **GP Services**: Routine check-ups, non-urgent medical issues, and prescription renewals.
-            """)
+            
+            # 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
 
@@ -323,8 +451,16 @@ def create_demo():
         raise
 
 if __name__ == "__main__":
-    # Launch Gradio Interface
+    # Initialize environment
+    load_dotenv()
+    
+    # Set up Hugging Face login if token exists
+    hf_token = os.getenv("HF_TOKEN")
+    if hf_token:
+        login(token=hf_token)
+    
+    # Launch demo
     demo = create_demo()
-    demo.launch()
+    demo.launch(share=True)
+
 
-