Update app.py

app.py

@@ -261,82 +261,12 @@ EMERGENCY_SCENARIOS = {
     }
 }
 
-def analyze_medical_image(image):
-    """Analyze medical image using multiple classifiers"""
-    
-    detected_conditions = []
-    confidence_scores = {}
-    
-    try:
-        # Try wound classifier first
-        wound_results = wound_classifier(image, top_k=5)
-        
-        for result in wound_results:
-            label = result['label'].lower()
-            score = result['score']
-            
-            # Store all results with scores
-            confidence_scores[label] = score
-            
-            # Map wound types to Urdu emergencies
-            if score > 0.2:
-                if any(word in label for word in ['burn', 'scald', 'thermal', 'fire']):
-                    detected_conditions.append(("جل جانا", score))
-                elif any(word in label for word in ['cut', 'laceration', 'wound', 'bleeding', 'incision']):
-                    detected_conditions.append(("خون بہنا", score))
-                elif any(word in label for word in ['abrasion', 'scrape', 'graze', 'scratch']):
-                    detected_conditions.append(("خون بہنا", score))
-                elif any(word in label for word in ['bruise', 'contusion', 'hematoma']):
-                    detected_conditions.append(("چوٹ یا نیلا نشان", score))
-                elif any(word in label for word in ['fracture', 'broken', 'bone']):
-                    detected_conditions.append(("ہڈی ٹوٹنا", score))
-                elif any(word in label for word in ['bite', 'animal', 'insect']):
-                    detected_conditions.append(("سانپ کا کاٹنا", score))
-                elif any(word in label for word in ['rash', 'skin', 'allergy', 'dermatitis']):
-                    detected_conditions.append(("جلد پر خارش یا دانے", score))
-                elif any(word in label for word in ['swelling', 'inflammation', 'edema']):
-                    detected_conditions.append(("چوٹ یا نیلا نشان", score))
-        
-        # If no specific wound detected, try general classifier
-        if not detected_conditions:
-            general_results = medical_classifier(image, top_k=5)
-            
-            for result in general_results:
-                label = result['label'].lower()
-                score = result['score']
-                
-                if score > 0.15:
-                    if any(word in label for word in ['wound', 'injury', 'medical', 'blood', 'skin', 'burn', 'cut']):
-                        detected_conditions.append(("خون بہنا", score))
-                        break
-        
-    except Exception as e:
-        print(f"Image analysis error: {e}")
-        return None, None
-    
-    # Return the most confident detection
-    if detected_conditions:
-        detected_conditions.sort(key=lambda x: x[1], reverse=True)
-        best_match = detected_conditions[0]
-        
-        # Create detailed analysis message
-        analysis_msg = f"\n\n📊 تصویر کی تجزیہ:\n"
-        analysis_msg += f"• پتہ چلی حالت: {best_match[0]}\n"
-        analysis_msg += f"• اعتماد کی سطح: {best_match[1]:.1%}\n"
-        
-        if len(detected_conditions) > 1:
-            analysis_msg += f"\n🔍 دیگر ممکنہ حالات:\n"
-            for condition, score in detected_conditions[1:3]:
-                analysis_msg += f"• {condition} ({score:.1%})\n"
-        
-        return best_match[0], analysis_msg
-    
-    return None, None
-
 def text_to_speech_urdu(text):
     """Convert Urdu text to speech"""
     try:
-        tts = gTTS(text=text, lang='ur', slow=False)
+        # Remove asterisks and other markdown formatting before TTS
+        clean_text = text.replace('*', '').replace('#', '').replace('•', '')
+        tts = gTTS(text=clean_text, lang='ur', slow=False)
         temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.mp3')
         tts.save(temp_file.name)
         return temp_file.name
@@ -344,74 +274,19 @@ def text_to_speech_urdu(text):
         print(f"TTS Error: {e}")
         return None
 
-def get_emergency_help(query_text, image=None):
+def get_emergency_help(scenario_name):
     """Get emergency first aid instructions"""
     
-    matched_emergency = None
-    image_analysis_msg = ""
-    
-    # First try image analysis if image is provided
-    if image is not None:
-        detected_emergency, analysis_msg = analyze_medical_image(image)
-        if detected_emergency:
-            matched_emergency = detected_emergency
-            image_analysis_msg = analysis_msg
-    
-    # If no match from image, try text query
-    if not matched_emergency and query_text:
-        query_lower = query_text.lower()
-        
-        keywords_map = {
-            "دل کا دورہ": ["دل", "heart", "chest", "سینہ", "سینے"],
-            "دم گھٹنا": ["گھٹ", "chok", "گلا", "throat", "سانس"],
-            "خون بہنا": ["خون", "blood", "کٹ", "cut", "زخم", "bleeding"],
-            "بے ہوشی": ["بے ہوش", "unconscious", "faint", "گر"],
-            "جل جانا": ["جل", "burn", "آگ", "fire", "جلنا"],
-            "ہڈی ٹوٹنا": ["ہڈی", "bone", "fracture", "توڑ", "ٹوٹ"],
-            "زہر کھا لینا": ["زہر", "poison", "کھا لیا"],
-            "دورہ پڑنا": ["دورہ", "seizure", "جھٹکے"],
-            "لو لگنا": ["لو", "heat", "گرمی"],
-            "سانپ کا کاٹنا": ["سانپ", "snake", "کاٹ", "bite"],
-            "دمہ کا دورہ": ["دمہ", "asthma", "سانس", "breath"],
-            "الرجی کا شدید ردعمل": ["الرجی", "allergy", "سوجن", "swell"],
-            "فالج": ["فالج", "stroke", "چہرہ", "face"],
-            "شوگر کم ہونا": ["شوگر", "sugar", "low", "کانپنا"],
-            "بجلی کا جھٹکا": ["بجلی", "electric", "shock", "جھٹکا"],
-            "ڈوبنا": ["ڈوب", "drown", "پانی", "water"],
-            "ناک سے خون بہنا": ["ناک", "nose", "bleed"],
-            "موچ": ["موچ", "sprain", "ٹخنہ", "ankle"],
-            "جلد پر خارش یا دانے": ["خارش", "rash", "دانے", "جلد", "skin"],
-            "چوٹ یا نیلا نشان": ["چوٹ", "bruise", "نیلا", "چوٹ"]
-        }
-        
-        for emergency, keywords in keywords_map.items():
-            if any(keyword in query_lower for keyword in keywords):
-                matched_emergency = emergency
-                break
-    
-    if not matched_emergency:
-        response = "میں نے آپ کی بات یا تصویر کو پہچان نہیں سکا۔ "
-        
-        if image is not None:
-            response += "\n\n💡 تجویز: براہ کرم:\n"
-            response += "• تصویر کو واضح اور روشنی میں لیں\n"
-            response += "• زخم یا متاثرہ حصہ قریب سے دکھائیں\n"
-            response += "• یا نیچے سے اپنی ایمرجنسی منتخب کریں\n\n"
-        
-        response += "براہ کرم مندرجہ ذیل میں سے منتخب کریں:\n\n"
-        response += "\n".join([f"• {key}" for key in EMERGENCY_SCENARIOS.keys()])
+    if scenario_name not in EMERGENCY_SCENARIOS:
+        response = "براہ کرم صحیح ایمرجنسی منتخب کریں"
         audio_file = text_to_speech_urdu(response)
         return response, audio_file
     
     # Get emergency info
-    emergency_info = EMERGENCY_SCENARIOS[matched_emergency]
+    emergency_info = EMERGENCY_SCENARIOS[scenario_name]
     
     # Format response without asterisks
-    response = f"🚨 {matched_emergency}\n\n"
-    
-    if image_analysis_msg:
-        response += image_analysis_msg + "\n"
-    
+    response = f"🚨 {scenario_name}\n\n"
     response += f"علامات: {emergency_info['symptoms']}\n\n"
     response += "فوری اقدامات:\n\n"
     
@@ -438,8 +313,82 @@ with gr.Blocks(title="ایمرجنسی فرسٹ ایڈ AI", theme=gr.themes.Soft
     # 🚑 ایمرجنسی فرسٹ ایڈ AI
     ## آپ کا زندگی بچانے والا ساتھی - اردو میں
     
-    کیسے استعمال کریں:
-    1. نیچے سے اپنی ایمرجنسی منتخب کریں
-    2. فوری اقدامات اردو میں ملیں گے
-    3. آواز سنیں یا پڑھیں - آپ کی مرضی!
-    """)
+    نیچے سے اپنی ایمرجنسی منتخب کریں اور فوری ہدایات حاصل کریں
+    """)
+    
+    gr.Markdown("### ⚡ عام ایمرجنسیز - منتخب کریں")
+    
+    with gr.Row():
+        with gr.Column():
+            btn1 = gr.Button("💔 دل کا دورہ", size="lg")
+            btn2 = gr.Button("🫁 دم گھٹنا", size="lg")
+            btn3 = gr.Button("🩸 خون بہنا", size="lg")
+            btn10 = gr.Button("🫀 دمہ کا دورہ", size="lg")
+            btn11 = gr.Button("🤧 الرجی کا ردعمل", size="lg")
+            btn16 = gr.Button("🔴 جلد پر خارش یا دانے", size="lg")
+        
+        with gr.Column():
+            btn4 = gr.Button("😵 بے ہوشی", size="lg")
+            btn5 = gr.Button("🔥 جل جانا", size="lg")
+            btn6 = gr.Button("🦴 ہڈی ٹوٹنا", size="lg")
+            btn12 = gr.Button("🧠 فالج", size="lg")
+            btn13 = gr.Button("🍬 شوگر کم ہونا", size="lg")
+            btn17 = gr.Button("🟣 چوٹ یا نیلا نشان", size="lg")
+        
+        with gr.Column():
+            btn7 = gr.Button("☠️ زہر کھا لینا", size="lg")
+            btn8 = gr.Button("⚡ دورہ پڑنا", size="lg")
+            btn9 = gr.Button("🌡️ لو لگنا", size="lg")
+            btn14 = gr.Button("⚡ بجلی کا جھٹکا", size="lg")
+            btn15 = gr.Button("🌊 ڈوبنا", size="lg")
+            btn18 = gr.Button("🩹 موچ", size="lg")
+    
+    with gr.Row():
+        with gr.Column():
+            btn19 = gr.Button("🐍 سانپ کا کاٹنا", size="lg")
+        with gr.Column():
+            btn20 = gr.Button("👃 ناک سے خون بہنا", size="lg")
+    
+    quick_output = gr.Textbox(label="فرسٹ ایڈ ہدایات", lines=20, rtl=True)
+    quick_audio = gr.Audio(label="آواز میں سنیں", type="filepath")
+    
+    # Connect all buttons with proper lambda functions
+    btn1.click(fn=lambda: quick_scenario("دل کا دورہ"), inputs=[], outputs=[quick_output, quick_audio])
+    btn2.click(fn=lambda: quick_scenario("دم گھٹنا"), inputs=[], outputs=[quick_output, quick_audio])
+    btn3.click(fn=lambda: quick_scenario("خون بہنا"), inputs=[], outputs=[quick_output, quick_audio])
+    btn4.click(fn=lambda: quick_scenario("بے ہوشی"), inputs=[], outputs=[quick_output, quick_audio])
+    btn5.click(fn=lambda: quick_scenario("جل جانا"), inputs=[], outputs=[quick_output, quick_audio])
+    btn6.click(fn=lambda: quick_scenario("ہڈی ٹوٹنا"), inputs=[], outputs=[quick_output, quick_audio])
+    btn7.click(fn=lambda: quick_scenario("زہر کھا لینا"), inputs=[], outputs=[quick_output, quick_audio])
+    btn8.click(fn=lambda: quick_scenario("دورہ پڑنا"), inputs=[], outputs=[quick_output, quick_audio])
+    btn9.click(fn=lambda: quick_scenario("لو لگنا"), inputs=[], outputs=[quick_output, quick_audio])
+    btn10.click(fn=lambda: quick_scenario("دمہ کا دورہ"), inputs=[], outputs=[quick_output, quick_audio])
+    btn11.click(fn=lambda: quick_scenario("الرجی کا شدید ردعمل"), inputs=[], outputs=[quick_output, quick_audio])
+    btn12.click(fn=lambda: quick_scenario("فالج"), inputs=[], outputs=[quick_output, quick_audio])
+    btn13.click(fn=lambda: quick_scenario("شوگر کم ہونا"), inputs=[], outputs=[quick_output, quick_audio])
+    btn14.click(fn=lambda: quick_scenario("بجلی کا جھٹکا"), inputs=[], outputs=[quick_output, quick_audio])
+    btn15.click(fn=lambda: quick_scenario("ڈوبنا"), inputs=[], outputs=[quick_output, quick_audio])
+    btn16.click(fn=lambda: quick_scenario("جلد پر خارش یا دانے"), inputs=[], outputs=[quick_output, quick_audio])
+    btn17.click(fn=lambda: quick_scenario("چوٹ یا نیلا نشان"), inputs=[], outputs=[quick_output, quick_audio])
+    btn18.click(fn=lambda: quick_scenario("موچ"), inputs=[], outputs=[quick_output, quick_audio])
+    btn19.click(fn=lambda: quick_scenario("سانپ کا کاٹنا"), inputs=[], outputs=[quick_output, quick_audio])
+    btn20.click(fn=lambda: quick_scenario("ناک سے خون بہنا"), inputs=[], outputs=[quick_output, quick_audio])
+    
+    gr.Markdown("""
+    ---
+    ### ⚠️ اہم یاد دہانی:
+    - یہ AI صرف ابتدائی مدد کے لیے ہے
+    - شدید ایمرجنسی میں فوری 1122 کال کریں
+    - پیشہ ور طبی مشورے کا متبادل نہیں
+    
+    🏥 ایمرجنسی نمبرز:
+    - Rescue 1122: 1122
+    - Edhi Ambulance: 115
+    - Chippa Ambulance: 1020
+    
+    Made with ❤️ for Pakistan | Competition Project by Zain
+    """)
+
+# Launch the app
+if __name__ == "__main__":
+    app.launch()