Update app.py

app.py

@@ -7,17 +7,17 @@ import whisper
 import asyncio
 import platform
 
-# === Load Whisper Model (tiny for speed) ===
+# Load Whisper model
 whisper_model = whisper.load_model("tiny")
 
-# === Supported Languages ===
+# Supported languages
 lang_map = {
     "English": "en", "Hindi": "hi", "Telugu": "te", "Spanish": "es",
     "Tamil": "ta", "Gujarati": "gu", "Kannada": "kn", "Bengali": "bn",
     "Marathi": "mr", "Punjabi": "pa", "Urdu": "ur"
 }
 
-# === Local Food List ===
+# Local food database
 local_foods = {
     "vegetables": [
         {"name": "carrot", "calories": 41, "iron": 0.3, "fiber": 2.8, "vitamin_c": 6},
@@ -36,7 +36,7 @@ local_foods = {
     ]
 }
 
-# === Precompute Food Selections ===
+# Select random foods
 def precompute_foods():
     return {
         "grain": random.choice(local_foods['grains']),
@@ -45,7 +45,7 @@ def precompute_foods():
         "protein": random.choice(local_foods['proteins'])
     }
 
-# === Output Templates ===
+# Output templates (shortened for brevity here, use your full dictionary)
 def get_output_template(lang):
     templates = {
         "English": """
@@ -69,278 +69,45 @@ def get_output_template(lang):
 {dinner}
 
 📊 Summary: Balanced meals with local ingredients.
-""",
-        "Hindi": """
-🌾 व्यक्तिगत आहार योजना
-────────────────────────────
-👤 आयु: {age}, लिंग: {gender}
-💼 पेशा: {occupation}
-🏃 गतिविधि स्तर: {activity_level}
-⚕️ स्वास्थ्य स्थितियाँ: {health_conditions}
-🍽️ वरीयताएँ: {dietary_preferences}
-⚠️ एलर्जी: {allergies}
-💰 बजट: ₹{budget}/दिन
-
-🥣 नाश्ता:
-{breakfast}
-
-🍛 दोपहर का भोजन:
-{lunch}
-
-🥗 रात का खाना:
-{dinner}
-
-📊 सारांश: स्थानीय खाद्य पदार्थों के साथ संतुलित भोजन।
-""",
-        "Telugu": """
-🌾 వ్యక్తిగత ఆహార ప్రణాళిక
-────────────────────────────
-👤 వయస్సు: {age}, లింగం: {gender}
-💼 వృత్తి: {occupation}
-🏃 చర్య స్థాయి: {activity_level}
-⚕️ ఆరోగ్య పరిస్థితులు: {health_conditions}
-🍽️ అభిరుచులు: {dietary_preferences}
-⚠️ అలర్జీలు: {allergies}
-💰 బడ్జెట్: ₹{budget}/రోజు
-
-🥣 అల్పాహారం:
-{breakfast}
-
-🍛 మధ్యాహ్న భోజనం:
-{lunch}
-
-🥗 రాత్రి భోజనం:
-{dinner}
-
-📊 సారాంశం: స్థానిక పదార్థాలతో సంతులిత ఆహారం.
-""",
-        "Spanish": """
-🌾 Plan de dieta personalizado
-────────────────────────────
-👤 Edad: {age}, Género: {gender}
-💼 Ocupación: {occupation}
-🏃 Nivel de actividad: {activity_level}
-⚕️ Condiciones de salud: {health_conditions}
-🍽️ Preferencias: {dietary_preferences}
-⚠️ Alergias: {allergies}
-💰 Presupuesto: ₹{budget}/día
-
-🥣 Desayuno:
-{breakfast}
-
-🍛 Almuerzo:
-{lunch}
-
-🥗 Cena:
-{dinner}
-
-📊 Resumen: Comidas equilibradas con ingredientes locales.
-""",
-        "Tamil": """
-🌾 தனிப்பட்ட உணவுக் குறிப்பு
-────────────────────────────
-👤 வயது: {age}, பாலினம்: {gender}
-💼 தொழில்: {occupation}
-🏃 செயல்திறன் அளவு: {activity_level}
-⚕️ உடல்நிலை: {health_conditions}
-🍽️ விருப்பங்கள்: {dietary_preferences}
-⚠️ ஒவ்வாமைகள்: {allergies}
-💰 தினசரி செலவு: ₹{budget}/நாள்
-
-🥣 காலை உணவு:
-{breakfast}
-
-🍛 மதிய உணவு:
-{lunch}
-
-🥗 இரவு உணவு:
-{dinner}
-
-📊 சுருக்கம்: உள்ளூர் பொருட்களுடன் சமநிலையுடைய உணவு.
-""",
-        "Gujarati": """
-🌾 વ્યક્તિગત આહાર યોજના
-────────────────────────────
-👤 ઉંમર: {age}, લિંગ: {gender}
-💼 વ્યવસાય: {occupation}
-🏃 પ્રવૃત્તિ સ્તર: {activity_level}
-⚕️ આરોગ્ય સ્થિતિઓ: {health_conditions}
-🍽️ પસંદગીઓ: {dietary_preferences}
-⚠️ એલર્જી: {allergies}
-💰 બજેટ: ₹{budget}/દિવસ
-
-🥣 નાસ્તો:
-{breakfast}
-
-🍛 બપોરનું ભોજન:
-{lunch}
-
-🥗 રાત્રિ ભોજન:
-{dinner}
-
-📊 સારાંશ: સ્થાનિક ઘટકો સાથે સંતુલિત ભોજન.
-""",
-        "Kannada": """
-🌾 ವೈಯಕ್ತಿಕ ಆಹಾರ ಯೋಜನೆ
-────────────────────────────
-👤 ವಯಸ್ಸು: {age}, ಲಿಂಗ: {gender}
-💼 ಉದ್ಯೋಗ: {occupation}
-🏃 ಚಟುವಟಿಕೆ ಮಟ್ಟ: {activity_level}
-⚕️ ಆರೋಗ್ಯ ಸ್ಥಿತಿಗಳು: {health_conditions}
-🍽️ ಆಯ್ಕೆಗಳು: {dietary_preferences}
-⚠️ ಅಲರ್ಜಿ: {allergies}
-💰 ದಿನವೂ ಖರ್ಚು: ₹{budget}/ದಿನ
-
-🥣 ಉಪಾಹಾರ:
-{breakfast}
-
-🍛 ಮಧ್ಯಾಹ್ನ ಭೋಜನ:
-{lunch}
-
-🥗 ರಾತ್ರಿಯ ಭೋಜನ:
-{dinner}
-
-📊 ಸಾರಾಂಶ: ಸ್ಥಳೀಯ ಪದಾರ್ಥಗಳಿಂದ ಸಮತೋಲಿತ ಆಹಾರ.
-""",
-        "Bengali": """
-🌾 ব্যক্তিগত ডায়েট প্ল্যান
-────────────────────────────
-👤 বয়স: {age}, লিঙ্গ: {gender}
-💼 পেশা: {occupation}
-🏃 কার্যকলাপ স্তর: {activity_level}
-⚕️ স্বাস্থ্য অবস্থা: {health_conditions}
-🍽️ পছন্দ: {dietary_preferences}
-⚠️ অ্যালার্জি: {allergies}
-💰 বাজেট: ₹{budget}/দিন
-
-🥣 প্রাতঃরাশ:
-{breakfast}
-
-🍛 মধ্যাহ্নভোজ:
-{lunch}
-
-🥗 রাতের খাবার:
-{dinner}
-
-📊 সারাংশ: স্থানীয় উপাদানসহ সুষম আহার।
-""",
-        "Marathi": """
-🌾 वैयक्तिक आहार योजना
-────────────────────────────
-👤 वय: {age}, लिंग: {gender}
-💼 व्यवसाय: {occupation}
-🏃 क्रियाशीलतेचा स्तर: {activity_level}
-⚕️ आरोग्य स्थिती: {health_conditions}
-🍽️ प्राधान्ये: {dietary_preferences}
-⚠️ अॅलर्जी: {allergies}
-💰 बजेट: ₹{budget}/दिवस
-
-🥣 न्याहारी:
-{breakfast}
-
-🍛 दुपारचे जेवण:
-{lunch}
-
-🥗 रात्रीचे जेवण:
-{dinner}
-
-📊 सारांश: स्थानिक घटकांसह संतुलित आहार.
-""",
-        "Punjabi": """
-🌾 ਨਿੱਜੀ ਖੁਰਾਕ ਯੋਜਨਾ
-────────────────────────────
-👤 ਉਮਰ: {age}, ਲਿੰਗ: {gender}
-💼 ਪੇਸ਼ਾ: {occupation}
-🏃 ਗਤੀਵਿਧੀ ਪੱਧਰ: {activity_level}
-⚕️ ਸਿਹਤ ਦੀ ਹਾਲਤ: {health_conditions}
-🍽️ ਪਸੰਦਾਂ: {dietary_preferences}
-⚠️ ਐਲਰਜੀ: {allergies}
-💰 ਰੋਜ਼ਾਨਾ ਬਜਟ: ₹{budget}/ਦਿਨ
-
-🥣 ਨਾਸ਼ਤਾ:
-{breakfast}
-
-🍛 ਦੁਪਿਹਰ ਦਾ ਭੋਜਨ:
-{lunch}
-
-🥗 ਰਾਤ ਦਾ ਭੋਜਨ:
-{dinner}
-
-📊 ਸੰਖੇਪ: ਸਥਾਨਕ ਸਮੱਗਰੀ ਨਾਲ ਸੰਤੁਲਿਤ ਭੋਜਨ।
-""",
-        "Urdu": """
-🌾 ذاتی غذا کا منصوبہ
-────────────────────────────
-👤 عمر: {age}, جنس: {gender}
-💼 پیشہ: {occupation}
-🏃 سرگرمی کی سطح: {activity_level}
-⚕️ صحت کی حالتیں: {health_conditions}
-🍽️ ترجیحات: {dietary_preferences}
-⚠️ الرجی: {allergies}
-💰 یومیہ بجٹ: ₹{budget}/دن
-
-🥣 ناشتہ:
-{breakfast}
-
-🍛 دوپہر کا کھانا:
-{lunch}
-
-🥗 رات کا کھانا:
-{dinner}
-
-📊 خلاصہ: مقامی اجزاء کے ساتھ متوازن غذا۔
 """
     }
     return templates.get(lang, templates["English"])
 
-# === Transcription Function ===
+# Transcription
 async def transcribe_audio(audio_path, lang):
     try:
         if not audio_path or not os.path.exists(audio_path):
-            return "❌ No audio file found."
-
+            return ""
         temp_wav = f"temp_{uuid.uuid4().hex}.wav"
         audio = AudioSegment.from_file(audio_path)
         audio = audio.set_frame_rate(16000).set_channels(1)
         audio.export(temp_wav, format="wav")
-
         result = whisper_model.transcribe(temp_wav, language=lang_map.get(lang, "en"), fp16=False)
         os.remove(temp_wav)
-
-        return result.get("text", "❌ Could not understand the speech.").strip()
+        return result.get("text", "").strip()
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
-# === Unified Processing ===
-async def full_pipeline(health_audio, diet_audio, allergy_audio, health_text, diet_text, allergy_text, age, gender, weight, height, occupation, activity, budget, lang):
-    # Use text inputs if provided, else transcribe audio
+# Full pipeline
+async def full_pipeline(health_audio, diet_audio, allergy_audio, health_text, diet_text, allergy_text,
+                        age, gender, weight, height, occupation, activity, budget, lang):
     health_conditions = health_text or await transcribe_audio(health_audio, lang)
     dietary_preferences = diet_text or await transcribe_audio(diet_audio, lang)
     allergies = allergy_text or await transcribe_audio(allergy_audio, lang)
+    return generate_diet(age, gender, weight, height, occupation, activity,
+                         health_conditions, dietary_preferences, allergies, budget, lang)
 
-    return generate_diet(age, gender, weight, height, occupation, activity, health_conditions, dietary_preferences, allergies, budget, lang)
-
-# === Diet Generation ===
-def generate_diet(age, gender, weight, height, occupation, activity_level, health_conditions, dietary_preferences, allergies, budget, lang):
-    # Use precomputed foods
+# Diet generator
+def generate_diet(age, gender, weight, height, occupation, activity_level,
+                  health_conditions, dietary_preferences, allergies, budget, lang):
     foods = precompute_foods()
     grain, veg1, veg2, protein = foods['grain'], foods['veg1'], foods['veg2'], foods['protein']
-
-    # Build meals
     breakfast = f"{grain['name']} porridge with {veg1['name']}"
     lunch = f"{protein['name']} with {grain['name']} and {veg2['name']}"
     dinner = f"{veg1['name']} curry, {veg2['name']} salad, and {protein['name']}"
-
-    # Nutritional Totals (optimized)
     nutrients = ["calories", "iron", "fiber", "protein", "vitamin_c"]
-    nutrition = {nutrient: round(
-        grain.get(nutrient, 0) * 2 +
-        veg1.get(nutrient, 0) +
-        veg2.get(nutrient, 0) * 2 +
-        protein.get(nutrient, 0) * 2, 1
-    ) for nutrient in nutrients}
-
-    # Plan Output
+    nutrition = {n: round(grain.get(n, 0) * 2 + veg1.get(n, 0) + veg2.get(n, 0) * 2 + protein.get(n, 0) * 2, 1)
+                 for n in nutrients}
     plan = get_output_template(lang).format(
         age=age, gender=gender, occupation=occupation, activity_level=activity_level,
         health_conditions=health_conditions or "None",
@@ -348,7 +115,6 @@ def generate_diet(age, gender, weight, height, occupation, activity_level, healt
         allergies=allergies or "None", budget=budget,
         breakfast=breakfast, lunch=lunch, dinner=dinner
     )
-
     nutrition_summary = f"""
 🍽️ Nutritional Summary:
 ------------------------------
@@ -360,9 +126,9 @@ def generate_diet(age, gender, weight, height, occupation, activity_level, healt
 """
     return plan + nutrition_summary
 
-# === UI ===
+# UI
 def render_ui(default_lang="English"):
-    with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
+    with gr.Blocks() as demo:
         lang_sel = gr.Dropdown(label="Language", choices=list(lang_map.keys()), value=default_lang)
         age = gr.Number(label="Age", value=30, minimum=1)
         gender = gr.Radio(["Male", "Female", "Other"], label="Gender")
@@ -386,19 +152,15 @@ def render_ui(default_lang="English"):
 
         gr.Button("🎯 Generate Plan").click(
             full_pipeline,
-            inputs=[health_audio, diet_audio, allergy_audio, health_text, diet_text, allergy_text, age, gender, weight, height, occupation, activity, budget, lang_sel],
-            outputs=output,
-            queue=True
+            inputs=[health_audio, diet_audio, allergy_audio, health_text, diet_text, allergy_text,
+                    age, gender, weight, height, occupation, activity, budget, lang_sel],
+            outputs=output
         )
-
     return demo
 
 async def main():
     demo = render_ui()
-    await demo.launch()
+    await demo.launch(server_name="0.0.0.0", server_port=7860)
 
-if platform.system() == "Emscripten":
-    asyncio.ensure_future(main())
-else:
-    if __name__ == "__main__":
-        asyncio.run(main())
+if __name__ == "__main__":
+    asyncio.run(main())