Create app.py

app.py

@@ -0,0 +1,134 @@
+import os
+import re
+import json
+import asyncio
+import tempfile
+
+import gradio as gr
+from telethon import TelegramClient
+from dotenv import load_dotenv
+
+# ================== LOAD ENV ==================
+load_dotenv()
+
+# ================== CONFIG ==================
+BOT_USERNAME = "mealschatbot"
+
+api_id = int(os.getenv("TG_API_ID"))
+api_hash = os.getenv("TG_API_HASH")
+phone_number = os.getenv("TG_PHONE")
+
+# ================== PARSER ==================
+def parse_food_message(text: str):
+    if "Calories" not in text:
+        return None
+
+    data = {}
+
+    # Item name
+    name_match = re.search(r"You added:\s*\*\*(.+?)\*\*", text)
+    if name_match:
+        data["item"] = name_match.group(1).strip()
+
+    # Calories
+    cal_match = re.search(
+        r"\*\*Calories\*\*\s*\n+\s*(\d+)\s*kcal",
+        text,
+        re.IGNORECASE,
+    )
+    if cal_match:
+        data["calories_kcal"] = int(cal_match.group(1))
+
+    # Macros
+    macros = {}
+    for key, pattern in {
+        "protein_g": r"Protein:\s*(\d+)g",
+        "carbs_g": r"Carbs:\s*(\d+)g",
+        "fat_g": r"Fat:\s*(\d+)g",
+    }.items():
+        m = re.search(pattern, text)
+        if m:
+            macros[key] = int(m.group(1))
+
+    if macros:
+        data["macros"] = macros
+
+    # Ingredients
+    ingredients = []
+    block = re.search(
+        r"\*\*Likely Ingredients\*\*\s*\n+([\s\S]+?)(?:📊|$)",
+        text,
+    )
+
+    if block:
+        for line in block.group(1).splitlines():
+            line = line.strip("• ").strip()
+            if not line:
+                continue
+
+            m = re.match(r"(.+?)\s*\((\d+)g,\s*(\d+)kcal\)", line)
+            if m:
+                ingredients.append({
+                    "name": m.group(1),
+                    "quantity_g": int(m.group(2)),
+                    "calories_kcal": int(m.group(3)),
+                })
+
+    if ingredients:
+        data["likely_ingredients"] = ingredients
+
+    return data
+
+
+# ================== TELEGRAM HANDLER ==================
+async def analyze_image(image_path: str):
+    async with TelegramClient("session", api_id, api_hash) as client:
+        await client.start(phone=phone_number)
+
+        bot = await client.get_entity(BOT_USERNAME)
+
+        history = await client.get_messages(bot, limit=1)
+        last_id_before = history[0].id if history else 0
+
+        await client.send_file(bot, image_path)
+        await asyncio.sleep(8)
+
+        replies = await client.get_messages(
+            bot,
+            min_id=last_id_before,
+            limit=10,
+        )
+
+        for msg in reversed(replies):
+            if msg.text:
+                parsed = parse_food_message(msg.text)
+                if parsed:
+                    return parsed
+
+    return {"error": "No food data detected"}
+
+
+# ================== GRADIO WRAPPER ==================
+def gradio_handler(image):
+    with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as f:
+        image.save(f.name)
+        result = asyncio.run(analyze_image(f.name))
+        return json.dumps(result, indent=2)
+
+
+# ================== UI ==================
+with gr.Blocks(title="Meal Image → Nutrition JSON") as demo:
+    gr.Markdown("## 🍽️ Meal Image → Nutrition JSON")
+    gr.Markdown("Upload a food image. Output will be **pure JSON only**.")
+
+    image_input = gr.Image(type="pil", label="Upload Food Image")
+    json_output = gr.Code(label="Parsed JSON", language="json")
+
+    analyze_btn = gr.Button("Analyze")
+    analyze_btn.click(
+        fn=gradio_handler,
+        inputs=image_input,
+        outputs=json_output,
+    )
+
+demo.launch()