Update app.py

app.py

@@ -4,7 +4,7 @@ import torch
 import re
 
 # =======================================================
-# Model Loading with Error Handling
+# Model Loading
 # =======================================================
 model_name = "augtoma/qCammel-13"
 print("Loading tokenizer and model...")
@@ -28,59 +28,33 @@ except Exception as e:
     raise
 
 
-# =======================================================
-# Constants
-# =======================================================
-MEDICAL_DISCLAIMER = """
-⚠️ **IMPORTANT MEDICAL DISCLAIMER:**
-- This is an AI simulation for educational purposes only
-- For medical emergencies, call emergency services immediately (911/999/local emergency number)
-- Always consult a licensed healthcare provider for diagnosis and treatment
-- Do not use this for serious, worsening, or life-threatening symptoms
-- This AI cannot diagnose conditions or prescribe medications
-"""
-
-EMERGENCY_KEYWORDS = [
-    "chest pain", "can't breathe", "breathing difficulty", "suicide", 
-    "heart attack", "stroke", "severe bleeding", "unconscious", 
-    "overdose", "severe pain", "emergency"
-]
-
-
 # =======================================================
 # Helper Functions
 # =======================================================
 def extract_name(text):
-    """Extract name from user input"""
     text = text.lower()
-    # Remove common phrases
-    for phrase in ["yes", "i am", "i'm", "my name is", "name is", "it's", "its"]:
+    for phrase in ["yes", "i am", "i'm", "my name is", "name is", "it's", "its", "this is"]:
         text = text.replace(phrase, "")
-    
-    # Clean and capitalize
     text = re.sub(r'[^\w\s]', '', text).strip()
     return text.title() if text and len(text) > 1 else "Patient"
 
 
 def extract_age(text):
-    """Extract age from user input"""
-    # Look for numbers in the text
     numbers = re.findall(r'\b(\d{1,3})\b', text)
     for num in numbers:
         age = int(num)
-        if 1 <= age <= 120:  # Reasonable age range
+        if 1 <= age <= 120:
             return age
     return None
 
 
 def check_emergency(text):
-    """Check if message contains emergency keywords"""
-    text_lower = text.lower()
-    return any(keyword in text_lower for keyword in EMERGENCY_KEYWORDS)
+    emergency_words = ["chest pain", "can't breathe", "suicide", "heart attack", "stroke", 
+                       "severe bleeding", "unconscious", "overdose"]
+    return any(word in text.lower() for word in emergency_words)
 
 
 def create_initial_session():
-    """Create a fresh session dictionary"""
     return {
         "name": None,
         "age": None,
@@ -88,156 +62,46 @@ def create_initial_session():
         "symptoms": None,
         "duration": None,
         "medication": None,
-        "stage": "intro"
+        "other_info": [],
+        "stage": "intro",
+        "conversation_context": []
     }
 
 
 # =======================================================
-# Diet Suggestions
-# =======================================================
-def get_diet_suggestion(name, symptoms):
-    """Get personalized diet suggestions based on symptoms"""
-    symptoms_lower = symptoms.lower()
-    
-    if "fever" in symptoms_lower:
-        return f"""🍎 **Diet Plan for {name} (Fever)**
-
-**Meals:**
-- 🥣 **Breakfast:** Oatmeal or boiled egg with banana/apple
-- 🍲 **Lunch:** Rice with lentil soup or boiled vegetables
-- 🍛 **Dinner:** Light soup or grilled chicken with plain rice
-
-**Hydration:**
-- 💧 Drink 8–10 glasses of water daily
-- Coconut water or clear soups
-- Herbal teas (chamomile, ginger)
-
-**Avoid:**
-- 🚫 Fried, oily, or spicy foods
-- Heavy meals
-- Caffeinated drinks
-
-**Tips:** Eat small portions frequently to help your body recover."""
-
-    elif any(word in symptoms_lower for word in ["stomach", "vomit", "nausea", "diarrhea"]):
-        return f"""🥗 **Diet Plan for {name} (Stomach Issues)**
-
-**BRAT Diet:**
-- 🍌 Bananas
-- 🍚 Rice (plain white rice)
-- 🍎 Applesauce
-- 🍞 Toast (dry or with minimal butter)
-
-**Hydration:**
-- 💧 Sip water frequently
-- Oral rehydration solution (ORS)
-- Clear broths
-
-**Avoid:**
-- 🚫 Dairy products
-- Fried or spicy foods
-- Caffeine and alcohol
-- Raw vegetables
-
-**Tips:** Start with small amounts and gradually increase as you feel better."""
-
-    elif any(word in symptoms_lower for word in ["cold", "flu", "cough", "sore throat"]):
-        return f"""🍊 **Diet Plan for {name} (Cold/Flu)**
-
-**Immunity Boosters:**
-- 🍋 Citrus fruits (oranges, lemons)
-- 🍯 Honey with warm water
-- 🧄 Garlic (natural antibiotic)
-- 🧅 Onions
-
-**Meals:**
-- 🍲 Chicken soup (helps clear congestion)
-- ☕ Ginger or green tea
-- 🥣 Warm porridge or oatmeal
-
-**Hydration:**
-- 💧 Plenty of warm fluids
-- Herbal teas
-
-**Avoid:**
-- 🚫 Sugary drinks
-- Chilled beverages
-- Dairy (may increase mucus)
-
-**Tips:** Stay warm and get plenty of rest."""
-
-    elif any(word in symptoms_lower for word in ["headache", "migraine"]):
-        return f"""🧘 **Diet Plan for {name} (Headache)**
-
-**Helpful Foods:**
-- 💧 Stay well-hydrated (dehydration causes headaches)
-- 🥔 Magnesium-rich: bananas, nuts, spinach
-- 🐟 Omega-3: fish, walnuts
-- 🧂 Small amounts of salt if low blood pressure
-
-**Avoid:**
-- 🚫 Processed foods
-- Artificial sweeteners
-- Aged cheeses
-- Alcohol
-- Excessive caffeine
-
-**Tips:** Eat regular meals; skipping meals can trigger headaches."""
-
-    else:
-        return f"""🥗 **General Healthy Diet Plan for {name}**
-
-**Balanced Meals:**
-- 🍎 Fresh fruits and vegetables
-- 🍗 Lean proteins (chicken, fish, legumes)
-- 🍚 Whole grains (brown rice, oats)
-- 🥜 Nuts and seeds
-
-**Hydration:**
-- 💧 8–10 glasses of water daily
-
-**Avoid:**
-- 🚫 Processed and fried foods
-- Excessive sugar and salt
-- Junk food
-
-**Tips:** Eat at regular intervals and maintain portion control."""
-
-
-# =======================================================
-# AI Response Generation
+# Generate Natural Doctor Response
 # =======================================================
-def generate_ai_response(session, user_message):
-    """Generate AI consultation response using the model"""
+def generate_doctor_response(session, user_message):
+    """Generate a natural, conversational doctor response"""
     try:
-        # Build the prompt
-        prompt = f"""You are Dr. Aiden — a warm, caring, and professional doctor.
-You are consulting a {session['age']}-year-old {session['gender']} named {session['name']}.
+        # Build conversational context
+        context = "\n".join(session.get("conversation_context", [])[-3:])  # Last 3 exchanges
+        
+        prompt = f"""You are Dr. Aiden, a kind and experienced family doctor having a natural conversation with your patient.
 
-Patient details:
-- Symptoms: {session['symptoms']}
+Patient Profile:
+- Name: {session['name']}
+- Age: {session['age']} years old  
+- Gender: {session['gender']}
+- Chief Complaint: {session['symptoms']}
 - Duration: {session['duration']}
-- Current medications: {session['medication']}
+- Medications: {session['medication']}
 
-Now the patient says: "{user_message}"
+Recent conversation:
+{context}
 
-Respond as a real doctor would — empathetic, clear, and personalized.
-Include:
-1. Acknowledge their condition
-2. Possible causes
-3. Simple home remedies or OTC medicines (if safe)
-4. Diet, rest, and hydration tips
-5. When to visit a real doctor
-6. End with a reassuring tone"""
+Patient: {user_message}
 
-        # Tokenize and generate
-        inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=2048).to(model.device)
+Dr. Aiden: """
+
+        inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=1800).to(model.device)
         
         gen_cfg = GenerationConfig(
-            temperature=0.7,
-            top_p=0.9,
-            max_new_tokens=350,
-            repetition_penalty=1.1,
+            temperature=0.8,
+            top_p=0.92,
+            max_new_tokens=300,
+            repetition_penalty=1.15,
+            do_sample=True,
             pad_token_id=tokenizer.pad_token_id,
             eos_token_id=tokenizer.eos_token_id
         )
@@ -245,148 +109,277 @@ Include:
         with torch.no_grad():
             output = model.generate(**inputs, generation_config=gen_cfg)
 
-        # Decode the output
-        output_text = tokenizer.decode(output[0], skip_special_tokens=True)
-        
-        # Try to extract just the response part
-        # The model might repeat the prompt, so we try to find where the actual response starts
-        if "reassuring tone" in output_text:
-            parts = output_text.split("reassuring tone")
-            if len(parts) > 1:
-                output_text = parts[-1].strip()
+        response = tokenizer.decode(output[0], skip_special_tokens=True)
         
-        # If the output contains the prompt, try to split it
-        if user_message in output_text:
-            parts = output_text.split(user_message)
-            if len(parts) > 1:
-                output_text = parts[-1].strip()
+        # Extract just the doctor's response
+        if "Dr. Aiden:" in response:
+            response = response.split("Dr. Aiden:")[-1].strip()
+        elif "Patient:" in response:
+            response = response.split("Patient:")[0].strip()
         
         # Clean up
-        output_text = output_text.strip()
+        response = response.strip()
         
-        # If output is too short or empty, provide a fallback
-        if len(output_text) < 20:
-            output_text = f"Hello {session['name']}, I understand you have {session['symptoms']} for {session['duration']}. This could be due to several reasons. I recommend staying hydrated, getting plenty of rest, and monitoring your temperature. If your fever persists beyond 5 days or gets worse, please visit a doctor in person for proper examination."
+        # If response is too short or looks like it failed, give a fallback
+        if len(response) < 30 or response.startswith("You are Dr"):
+            response = generate_fallback_response(session, user_message)
         
-        # Ensure proper ending punctuation
-        if not output_text.endswith((".", "!", "?")):
-            output_text += "."
-
-        # Add disclaimer
-        output_text += "\n\n⚕️ *Note: This advice is AI-generated and not a substitute for professional medical care.*"
+        # Add to conversation context
+        session["conversation_context"].append(f"Patient: {user_message}")
+        session["conversation_context"].append(f"Doctor: {response}")
         
-        return output_text
+        return response
         
     except Exception as e:
-        print(f"Error in generate_ai_response: {e}")
-        return f"I understand you're experiencing {session['symptoms']}, {session['name']}. Since this has been going on for {session['duration']}, I recommend consulting with a healthcare provider for a proper examination and diagnosis.\n\n⚕️ *Note: This advice is AI-generated and not a substitute for professional medical care.*"
+        print(f"Generation error: {e}")
+        return generate_fallback_response(session, user_message)
+
+
+def generate_fallback_response(session, user_message):
+    """Generate a fallback response based on symptoms"""
+    name = session['name']
+    symptoms = session['symptoms'].lower()
+    duration = session['duration']
+    
+    if "fever" in symptoms:
+        return f"""I can see you've had this fever for {duration}, {name}. That's quite uncomfortable, I'm sure.
+
+A fever lasting 5 days could be viral or bacterial. Let me give you some advice:
+
+First, make sure you're drinking plenty of fluids - water, coconut water, or clear soups. Your body needs hydration when fighting infection.
+
+For the fever itself, you can take paracetamol (like Panadol) 500mg every 6 hours if the temperature goes above 100°F. 
+
+Get lots of rest - your body heals when you sleep. Eat light meals like khichdi, soup, or boiled vegetables.
+
+Now here's what concerns me: if your fever goes above 103°F, you develop severe headache, difficulty breathing, or if it continues beyond 7 days, please visit a clinic immediately. These could be signs of something more serious like dengue or typhoid, which need proper testing.
+
+How are you feeling otherwise? Any body aches, chills, or other symptoms I should know about?"""
+    
+    elif any(word in symptoms for word in ["cold", "cough", "flu"]):
+        return f"""Okay {name}, so you've been dealing with this for {duration}. Cold and flu can really drain your energy.
+
+Here's what I want you to do:
+
+Stay warm and rest as much as possible. Your immune system works best when you're sleeping.
+
+Drink warm liquids - ginger tea with honey is excellent. It soothes the throat and honey has antibacterial properties.
+
+For congestion, try steam inhalation 2-3 times a day. Just hot water in a bowl, put a towel over your head, and breathe in the steam for 5-10 minutes.
+
+You can take an antihistamine like Cetirizine at night if the symptoms are disturbing your sleep.
+
+Eat vitamin C rich foods - oranges, lemons, guavas. They boost immunity.
+
+Most colds resolve in 7-10 days. But if you develop high fever, chest pain, or difficulty breathing, that's when you need to see me in person or visit a doctor immediately.
+
+Are you able to eat properly? Any chest discomfort?"""
+    
+    else:
+        return f"""I understand you're concerned about {session['symptoms']}, {name}. You've been dealing with this for {duration}, which I know must be worrying.
+
+Let me help you with this. Based on what you've told me, here are some general recommendations:
+
+1. Keep track of your symptoms - note if they're getting better or worse
+2. Stay well hydrated - aim for 8-10 glasses of water daily  
+3. Get adequate rest and avoid stress
+4. Eat balanced, nutritious meals
+
+However, {name}, I want to be honest with you - some conditions really need a physical examination and possibly some tests. If your symptoms are severe, getting worse, or not improving in a few days, please visit a healthcare facility.
+
+Is there anything specific that's worrying you the most? Or any other symptoms you haven't mentioned yet?"""
 
 
 # =======================================================
-# Doctor Response Logic
+# Main Response Logic
 # =======================================================
 def doctor_response(user_message, session):
-    """Generate doctor's response based on conversation stage"""
+    """Handle conversation flow naturally"""
     if not user_message or not user_message.strip():
-        return "Please type your message.", session
+        return "I'm here, go ahead.", session
     
     user_message_clean = user_message.strip()
     user_message_lower = user_message_clean.lower()
     
-    # Check for emergency situations
+    # Emergency check
     if check_emergency(user_message_clean):
-        emergency_response = """🚨 **EMERGENCY ALERT** 🚨
-
-Your message suggests a potentially serious medical emergency.
+        return """⚠️ I'm concerned about what you just told me. This sounds like it could be a medical emergency.
 
-**PLEASE TAKE IMMEDIATE ACTION:**
-- Call emergency services NOW (911 in US, 999 in UK, or your local emergency number)
-- Go to the nearest emergency room
-- Do not delay seeking immediate medical attention
+Please call emergency services RIGHT NOW:
+- Pakistan: 1122 or 115
+- Or go to the nearest emergency room immediately
 
-This AI chatbot cannot handle emergencies. Your safety is the priority."""
-        return emergency_response, session
+Don't wait. Your safety is what matters most. This chat cannot handle emergencies.""", session
     
-    # Stage-based conversation flow
+    # Conversation flow
     try:
         if session["stage"] == "intro":
             session["stage"] = "ask_name"
-            return "👨‍⚕️ Hello! I'm Dr. Aiden, your AI medical assistant.\n\n" + MEDICAL_DISCLAIMER + "\n\nMay I know your name, please?", session
+            return """Hello! Come in, come in. I'm Dr. Aiden.
+
+Please, have a seat. Before we begin, I just want you to know this is an AI consultation for guidance only. For serious conditions, you should see a doctor in person.
+
+Now, what's your name?""", session
         
         elif session["stage"] == "ask_name":
             session["name"] = extract_name(user_message_clean)
             session["stage"] = "ask_age"
-            return f"Nice to meet you, {session['name']}! How old are you?", session
+            return f"Nice to meet you, {session['name']}. And how old are you?", session
         
         elif session["stage"] == "ask_age":
             age = extract_age(user_message_clean)
             if age:
                 session["age"] = age
                 session["stage"] = "ask_gender"
-                return f"Got it, {session['name']}. Are you male or female?", session
-            else:
-                return "Please tell me your age in numbers (e.g., 25, 30, 45).", session
+                return f"Alright, {session['age']} years old. Are you male or female?", session
+            return "Sorry, I didn't catch your age. Could you tell me in numbers? Like 25 or 30?", session
         
         elif session["stage"] == "ask_gender":
             if "male" in user_message_lower and "female" not in user_message_lower:
                 session["gender"] = "male"
             elif "female" in user_message_lower:
-                session["gender"] = "female"
+                session["gender"] = "female"  
             else:
-                return "Could you please specify whether you are male or female?", session
+                return "Male or female?", session
             
             session["stage"] = "ask_symptoms"
-            return f"Thanks, {session['name']}! So you're a {session['age']}-year-old {session['gender']}.\n\nWhat symptoms are you experiencing? Please describe them in detail.", session
+            return f"Got it. So {session['name']}, tell me - what brings you here today? What's been bothering you?", session
         
         elif session["stage"] == "ask_symptoms":
             if len(user_message_clean) < 5:
-                return "Please describe your symptoms in more detail so I can help you better.", session
+                return "Can you describe it a bit more? I want to understand what you're going through.", session
             session["symptoms"] = user_message_clean
             session["stage"] = "ask_duration"
-            return "Since when have you been feeling this way? (e.g., 2 days, 1 week, etc.)", session
+            return "I see. And when did this start? How long have you been feeling this way?", session
         
         elif session["stage"] == "ask_duration":
             session["duration"] = user_message_clean
             session["stage"] = "ask_medicine"
-            return "Are you currently taking any medications or treatments? If not, just say 'no'.", session
+            return "Okay. Are you taking any medicines for this right now? Any tablets or syrups?", session
         
         elif session["stage"] == "ask_medicine":
-            session["medication"] = user_message_clean
+            session["medication"] = user_message_clean if "no" not in user_message_lower else "No current medications"
             session["stage"] = "consult"
-            # Provide summary and automatically generate initial consultation
-            summary = f"""Thank you for providing all the information, {session['name']}.
-
-📋 **Summary:**
-- **Name:** {session['name']}
-- **Age:** {session['age']} years old
-- **Gender:** {session['gender']}
-- **Symptoms:** {session['symptoms']}
-- **Duration:** {session['duration']}
-- **Current medication:** {session['medication']}
-
-Let me assess your condition...\n\n"""
             
-            # Generate initial consultation automatically
-            initial_consultation = generate_ai_response(session, "Please assess my condition and give me advice")
-            return summary + initial_consultation, session
+            # Natural transition to consultation
+            return f"""Alright {session['name']}, let me understand this properly.
+
+So you're {session['age']} years old, and you've been having {session['symptoms']} for about {session['duration']}. {"You're not taking anything for it right now" if "no" in user_message_lower else f"And you're currently taking {session['medication']}"}.
+
+Let me share my thoughts with you...\n\n""" + generate_doctor_response(session, "What do you think about my condition and what should I do?"), session
         
         elif session["stage"] == "consult":
-            # Handle diet-specific queries
-            if any(word in user_message_lower for word in ["diet", "food", "meal", "eat", "nutrition", "drink"]):
-                diet_plan = get_diet_suggestion(session["name"], session["symptoms"])
-                return diet_plan + "\n\n⚕️ *Note: This is general advice. Consult a healthcare provider for personalized dietary recommendations, especially if you have allergies or other conditions.*", session
+            # Natural ongoing consultation
+            
+            # Diet questions
+            if any(word in user_message_lower for word in ["diet", "food", "eat", "meal"]):
+                return get_diet_advice(session), session
+            
+            # Medicine questions  
+            if any(word in user_message_lower for word in ["medicine", "tablet", "drug", "medication", "panadol"]):
+                return get_medicine_advice(session), session
             
-            # Generate AI consultation response for any other question
-            return generate_ai_response(session, user_message_clean), session
+            # General questions
+            return generate_doctor_response(session, user_message_clean), session
         
-        else:
-            # Reset if stage is unknown
-            session["stage"] = "intro"
-            return "Let's start over. Say 'Hi Doctor' to begin your consultation.", session
-    
     except Exception as e:
-        print(f"Error in doctor_response: {e}")
-        return "I apologize, but I encountered an error. Please try again or start a new consultation.", session
+        print(f"Error: {e}")
+        return "Sorry, could you repeat that? I want to make sure I understand you correctly.", session
+
+
+def get_diet_advice(session):
+    """Natural diet advice"""
+    name = session['name']
+    symptoms = session['symptoms'].lower()
+    
+    if "fever" in symptoms:
+        return f"""Good question about diet, {name}. When you have a fever, your body needs energy but your digestion is weak.
+
+Here's what I recommend:
+
+**For breakfast**: Keep it light - maybe some oatmeal porridge or a boiled egg with a banana. Easy to digest.
+
+**Lunch and dinner**: Khichdi is excellent - rice and dal together. Or you can have plain rice with some boiled vegetables or light chicken soup.
+
+**Throughout the day**: This is very important - drink lots of fluids. Water, coconut water, lemon water. Your body loses water when you have fever.
+
+**Avoid these**: No fried foods, no spicy food, no heavy curries. Your stomach can't handle them right now.
+
+Little meals, multiple times a day. Don't force yourself to eat if you're not hungry, but definitely keep drinking fluids.
+
+Does that make sense? Any other questions about food?"""
+    
+    elif "cold" in symptoms or "cough" in symptoms:
+        return f"""Ah yes, diet is important for recovery, {name}.
+
+When you have cold and cough:
+
+**Hot is good**: Warm soups, especially chicken soup - it's not just comfort food, it actually helps with congestion. Ginger tea with honey is excellent.
+
+**Fruits**: Oranges, kinnow, any citrus fruits. They have vitamin C which helps your immune system fight the infection.
+
+**Regular meals**: Dal rice, roti sabzi - normal food is fine. Just make sure it's warm, not cold from the fridge.
+
+**Avoid**: Cold drinks, ice cream, banana (can increase mucus), dairy products if you have lot of phlegm.
+
+**Drink**: Warm water throughout the day. You can add a bit of honey and lemon to it.
+
+And one home remedy - at night before sleeping, have a glass of warm milk with half teaspoon of turmeric. It helps with throat irritation.
+
+Anything else you'd like to know?"""
+    
+    else:
+        return f"""For what you're experiencing, {name}, I'd say maintain a balanced diet.
+
+Eat fresh vegetables, fruits, whole grains. Include protein - dal, chicken, fish, eggs. Your body needs good nutrition to heal.
+
+Stay hydrated - very important. At least 8-10 glasses of water daily.
+
+Avoid processed foods, too much sugar, fried items.
+
+Eat at regular times, don't skip meals.
+
+Is there something specific about your diet you're concerned about?"""
+
+
+def get_medicine_advice(session):
+    """Natural medicine advice"""
+    name = session['name']
+    symptoms = session['symptoms'].lower()
+    
+    response = f"""Alright {name}, regarding medications:\n\n"""
+    
+    if "fever" in symptoms:
+        response += """For your fever, you can take **Paracetamol 500mg** (Panadol is the common brand here).
+
+Take one tablet every 6 hours, but only when your temperature is above 100°F. Don't take it more than 4 times in 24 hours.
+
+Take it after eating something, even if just a few biscuits. Don't take it on empty stomach.
+
+If the fever is not coming down with Paracetamol, or if it's going very high (above 103°F), that's a red flag. You need to see a doctor in person then."""
+    
+    elif "cold" in symptoms or "cough" in symptoms:
+        response += """For cold and cough:
+
+**For runny nose/sneezing**: Cetirizine 10mg, one tablet at night before sleeping. It'll help you rest better.
+
+**For cough**: If it's dry cough - any cough syrup with dextromethorphan. If you have phlegm - syrup with guaifenesin (expectorant). Take as per bottle instructions.
+
+**For blocked nose**: You can use saline nasal drops.
+
+These are all over-the-counter, you can get them from any pharmacy.
+
+But listen carefully - if you develop fever, chest pain, or difficulty breathing, stop these and consult a doctor immediately."""
+    
+    else:
+        response += """I can give you general guidance, but I'd be more comfortable if you saw a doctor in person for proper examination and prescription.
+
+Some conditions need specific tests or examinations before recommending medicines.
+
+For now, stay comfortable, rest well, and monitor your symptoms. If anything worsens, please visit a clinic."""
+    
+    response += "\n\nAny other questions for me?"
+    return response
 
 
 # =======================================================
@@ -394,19 +387,17 @@ Let me assess your condition...\n\n"""
 # =======================================================
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.HTML("""
-    <div style="text-align:center; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); color:white; padding:25px; border-radius:15px; box-shadow: 0 4px 6px rgba(0,0,0,0.1);">
-        <h1 style="margin:0; font-size: 2.5em;">🏥 Dr. Aiden - AI Medical Assistant</h1>
-        <p style="margin:10px 0 0 0; font-size: 1.1em;">Compassionate AI-powered health consultation</p>
-        <p style="margin:5px 0 0 0; font-size: 0.9em; opacity: 0.9;">⚠️ For emergencies, call 911 immediately</p>
+    <div style="text-align:center; background: linear-gradient(135deg, #2193b0 0%, #6dd5ed 100%); color:white; padding:30px; border-radius:15px;">
+        <h1 style="margin:0; font-size: 2.5em;">🩺 Dr. Aiden's Clinic</h1>
+        <p style="margin:10px 0 0 0; font-size: 1.2em;">Your friendly AI doctor - here to listen and help</p>
     </div>
     """)
     
-    # Session state
     session_state = gr.State(create_initial_session)
     
     chatbot = gr.Chatbot(
-        label="👨‍⚕️ Chat with Dr. Aiden",
-        height=500,
+        label="💬 Consultation Room",
+        height=550,
         type='messages',
         avatar_images=(
             "https://cdn-icons-png.flaticon.com/512/706/706830.png",
@@ -416,33 +407,26 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     
     with gr.Row():
         user_input = gr.Textbox(
-            placeholder="Type your message here (say 'Hi Doctor' to start)...", 
-            label="Your Message", 
+            placeholder="Type your message here...", 
+            label="", 
             lines=2,
             scale=4
         )
     
     with gr.Row():
         send_btn = gr.Button("💬 Send", variant="primary", scale=1)
-        clear_btn = gr.Button("🔄 New Consultation", variant="secondary", scale=1)
+        clear_btn = gr.Button("🔄 New Patient", variant="secondary", scale=1)
     
     gr.HTML("""
-    <div style="margin-top: 20px; padding: 15px; background-color: #fff3cd; border-left: 4px solid #ffc107; border-radius: 5px;">
-        <strong>⚠️ Important Reminders:</strong>
-        <ul style="margin: 10px 0; padding-left: 20px;">
-            <li>This is an AI simulation for educational purposes only</li>
-            <li>Always consult a real doctor for medical diagnosis and treatment</li>
-            <li>For emergencies, call 911 or your local emergency services immediately</li>
-            <li>Do not use this for serious or life-threatening conditions</li>
-        </ul>
+    <div style="margin-top: 15px; padding: 12px; background-color: #fff3cd; border-radius: 8px; font-size: 0.9em;">
+        <strong>⚠️ Important:</strong> This is an AI simulation. Always consult a real doctor for medical conditions. 
+        For emergencies call 1122 (Pakistan) or 911.
     </div>
     """)
     
     def respond(message, history, session):
-        """Handle user message and update chat"""
         if history is None:
             history = []
-        
         if not message or not message.strip():
             return "", history, session
         
@@ -454,31 +438,13 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         return "", history, updated_session
     
     def reset():
-        """Reset conversation and session"""
         return [], create_initial_session()
     
-    # Event handlers
-    send_btn.click(
-        respond, 
-        [user_input, chatbot, session_state], 
-        [user_input, chatbot, session_state]
-    )
-    
-    user_input.submit(
-        respond, 
-        [user_input, chatbot, session_state], 
-        [user_input, chatbot, session_state]
-    )
-    
-    clear_btn.click(
-        reset, 
-        None, 
-        [chatbot, session_state], 
-        queue=False
-    )
+    send_btn.click(respond, [user_input, chatbot, session_state], [user_input, chatbot, session_state])
+    user_input.submit(respond, [user_input, chatbot, session_state], [user_input, chatbot, session_state])
+    clear_btn.click(reset, None, [chatbot, session_state], queue=False)
 
 if __name__ == "__main__":
-    print("🏥 Launching Dr. Aiden Medical Assistant...")
-    print("⚠️  Remember: This is for educational purposes only!")
+    print("🩺 Dr. Aiden's Clinic is now open...")
     demo.queue()
     demo.launch(share=True)