Create app.py

app.py

@@ -0,0 +1,404 @@
+import os
+import random
+import uuid
+import gradio as gr
+from pydub import AudioSegment
+import whisper
+import asyncio
+import platform
+
+# === Load Whisper Model (tiny for speed) ===
+whisper_model = whisper.load_model("tiny")
+
+# === 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_foods = {
+    "vegetables": [
+        {"name": "carrot", "calories": 41, "iron": 0.3, "fiber": 2.8, "vitamin_c": 6},
+        {"name": "spinach", "calories": 23, "iron": 2.7, "fiber": 2.2, "vitamin_c": 28},
+        {"name": "beetroot", "calories": 43, "iron": 0.8, "fiber": 2.0, "vitamin_c": 4}
+    ],
+    "grains": [
+        {"name": "rice", "calories": 130, "iron": 1.2, "fiber": 0.4, "protein": 2.7},
+        {"name": "wheat", "calories": 340, "iron": 3.2, "fiber": 12.2, "protein": 13},
+        {"name": "millet", "calories": 119, "iron": 3.0, "fiber": 8.5, "protein": 3.5}
+    ],
+    "proteins": [
+        {"name": "lentils", "calories": 116, "iron": 3.3, "fiber": 7.9, "protein": 9},
+        {"name": "chicken", "calories": 239, "iron": 1.3, "fiber": 0.0, "protein": 27},
+        {"name": "tofu", "calories": 76, "iron": 1.6, "fiber": 0.3, "protein": 8}
+    ]
+}
+
+# === Precompute Food Selections ===
+def precompute_foods():
+    return {
+        "grain": random.choice(local_foods['grains']),
+        "veg1": random.choice(local_foods['vegetables']),
+        "veg2": random.choice(local_foods['vegetables']),
+        "protein": random.choice(local_foods['proteins'])
+    }
+
+# === Output Templates ===
+def get_output_template(lang):
+    templates = {
+        "English": """
+🌾 Personalized Diet Plan
+────────────────────────────
+👤 Age: {age}, Gender: {gender}
+💼 Occupation: {occupation}
+🏃 Activity Level: {activity_level}
+⚕️ Health Conditions: {health_conditions}
+🍽️ Preferences: {dietary_preferences}
+⚠️ Allergies: {allergies}
+💰 Budget: ₹{budget}/day
+
+🥣 Breakfast:
+{breakfast}
+
+🍛 Lunch:
+{lunch}
+
+🥗 Dinner:
+{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 ===
+async def transcribe_audio(audio_path, lang):
+    try:
+        if not audio_path or not os.path.exists(audio_path):
+            return "❌ No audio file found."
+
+        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()
+    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
+    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)
+
+# === Diet Generation ===
+def generate_diet(age, gender, weight, height, occupation, activity_level, health_conditions, dietary_preferences, allergies, budget, lang):
+    # Use precomputed foods
+    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
+    plan = get_output_template(lang).format(
+        age=age, gender=gender, occupation=occupation, activity_level=activity_level,
+        health_conditions=health_conditions or "None",
+        dietary_preferences=dietary_preferences or "None",
+        allergies=allergies or "None", budget=budget,
+        breakfast=breakfast, lunch=lunch, dinner=dinner
+    )
+
+    nutrition_summary = f"""
+🍽️ Nutritional Summary:
+------------------------------
+🔥 Calories: {nutrition['calories']} kcal
+🩸 Iron: {nutrition['iron']} mg
+🧬 Protein: {nutrition['protein']} g
+🌾 Fiber: {nutrition['fiber']} g
+🍊 Vitamin C: {nutrition['vitamin_c']} mg
+"""
+    return plan + nutrition_summary
+
+# === UI ===
+def render_ui(default_lang="English"):
+    with gr.Blocks(theme=gr.themes.Monochrome()) 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")
+        weight = gr.Number(label="Weight (kg)", value=70, minimum=1)
+        height = gr.Number(label="Height (cm)", value=170, minimum=1)
+        occupation = gr.Textbox(label="Occupation")
+        activity = gr.Radio(["Low", "Moderate", "High"], label="Activity Level")
+
+        with gr.Row():
+            health_audio = gr.Audio(label="🎤 Speak Health Conditions", type="filepath")
+            health_text = gr.Textbox(label="Or Type Health Conditions")
+        with gr.Row():
+            diet_audio = gr.Audio(label="🎤 Speak Dietary Preferences", type="filepath")
+            diet_text = gr.Textbox(label="Or Type Dietary Preferences")
+        with gr.Row():
+            allergy_audio = gr.Audio(label="🎤 Speak Allergies", type="filepath")
+            allergy_text = gr.Textbox(label="Or Type Allergies")
+
+        budget = gr.Number(label="Daily Budget (₹)", value=200, minimum=0)
+        output = gr.Textbox(label="Generated Diet Plan", lines=15)
+
+        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
+        )
+
+    return demo
+
+async def main():
+    demo = render_ui()
+    await demo.launch()
+
+if platform.system() == "Emscripten":
+    asyncio.ensure_future(main())
+else:
+    if __name__ == "__main__":
+        asyncio.run(main())