Update app.py

app.py

@@ -5,10 +5,10 @@ from phi.tools.duckduckgo import DuckDuckGo
 from dotenv import load_dotenv
 import os
 
-# Load environment
+# Load environment variables
 load_dotenv()
 
-# Agent definition
+# Setup agent
 nutrition_agent = Agent(
     name="nutrition_agent",
     model=Gemini(id="gemini-2.5-flash-preview-04-17"),
@@ -26,7 +26,7 @@ nutrition_agent = Agent(
     markdown=True
 )
 
-# Gradio input processing
+# Function to call agent
 def generate_diet(name, age, gender, diet_type, location, health_issues, goals, likes, dislikes, calories):
     user_input = (
         f"My name is {name}. I am {age} years old, {gender}, {diet_type}, living in {location}. "
@@ -36,29 +36,42 @@ def generate_diet(name, age, gender, diet_type, location, health_issues, goals,
         "Please generate a personalized meal plan broken into breakfast, lunch, snacks, dinner. "
         "Also include a clear explanation of nutritional content and a conclusion for health advice."
     )
-    return nutrition_agent.run(user_input)
 
-# Gradio interface
-iface = gr.Interface(
-    fn=generate_diet,
-    inputs=[
-        gr.Textbox(label="Name"),
-        gr.Textbox(label="Age"),
-        gr.Radio(["Male", "Female", "Other"], label="Gender"),
-        gr.Radio(["Vegetarian", "Non-Vegetarian"], label="Diet Type"),
-        gr.Textbox(label="Location"),
-        gr.Textbox(label="Health Issues"),
-        gr.Textbox(label="Fitness Goals"),
-        gr.Textbox(label="Foods You Like"),
-        gr.Textbox(label="Foods You Dislike"),
-        gr.Textbox(label="Daily Calorie Requirement"),
-    ],
-    outputs=gr.Markdown(label="🧠 Your Personalized Nutrition Plan"),
-    title="🥗 Smartest AI Nutrition Assistant",
-    description="Get a fully personalized diet plan with nutritional breakdown using AI.",
-    theme="soft"
-)
+    # Run the agent
+    result = nutrition_agent.run(user_input)
+
+    # Ensure output is readable
+    if hasattr(result, "content"):
+        return result.content
+    elif isinstance(result, str):
+        return result
+    else:
+        return "⚠️ Unexpected response format. Please try again."
+
+# Gradio UI
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("## 🥗 Smartest AI Nutrition Assistant")
+    with gr.Row():
+        with gr.Column():
+            name = gr.Textbox(label="Name")
+            age = gr.Textbox(label="Age")
+            gender = gr.Radio(["Male", "Female", "Other"], label="Gender")
+            diet_type = gr.Radio(["Vegetarian", "Non-Vegetarian"], label="Diet Type")
+            location = gr.Textbox(label="Location")
+            health_issues = gr.Textbox(label="Health Issues")
+            goals = gr.Textbox(label="Fitness Goals")
+            likes = gr.Textbox(label="Foods You Like")
+            dislikes = gr.Textbox(label="Foods You Dislike")
+            calories = gr.Textbox(label="Daily Calorie Requirement")
+            submit = gr.Button("Get Diet Plan")
+        with gr.Column():
+            output = gr.Markdown(label="🧠 Your Personalized Diet Plan")
+
+    submit.click(
+        fn=generate_diet,
+        inputs=[name, age, gender, diet_type, location, health_issues, goals, likes, dislikes, calories],
+        outputs=output
+    )
 
-# Launch the UI
 if __name__ == "__main__":
-    iface.launch()
+    demo.launch()