Update app.py

app.py

@@ -2,69 +2,64 @@ import gradio as gr
 import google.generativeai as genai
 import os
 
-# Get API key
-api_key = os.getenv("GOOGLE_API_KEY")
-
-if not api_key:
-    raise ValueError("API key missing")
-
-genai.configure(api_key=api_key)
+# Configure API key
+genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
+# Use correct model
 model = genai.GenerativeModel("gemini-1.5-flash")
 
 def fitness_coach(age, weight, goal, activity):
     try:
         prompt = f"""
-        Create a fitness plan:
-        Age: {age}
-        Weight: {weight}
-        Goal: {goal}
-        Activity: {activity}
-        """
+You are a professional fitness coach.
+
+Generate a UNIQUE and personalized fitness plan based on the user details.
+
+User Details:
+Age: {age}
+Weight: {weight} kg
+Goal: {goal}
+Activity Level: {activity}
+
+Instructions:
+- If goal is Weight Loss → focus on fat loss workouts
+- If goal is Muscle Gain → focus on strength training
+- If goal is Maintenance → balanced plan
+- If activity is Beginner → simple plan
+- If Intermediate → moderate plan
+- If Advanced → intense plan
+
+Include:
+1. Workout Plan
+2. Diet Plan
+3. Tips
+
+Make sure the response changes based on inputs.
+"""
 
-        response = model.generate_content(prompt)
+        response = model.generate_content(
+            prompt,
+            generation_config={
+                "temperature": 0.9,
+                "top_p": 1,
+                "top_k": 1,
+                "max_output_tokens": 500,
+            }
+        )
 
-        if response and hasattr(response, "text"):
-            return response.text
+        return response.text
 
     except Exception as e:
-        print("API failed:", e)
-
-    
-    return f"""
-💪 Personalized Fitness Plan
-
-👤 Age: {age}
-⚖ Weight: {weight} kg
-🎯 Goal: {goal}
-🏃 Activity Level: {activity}
-
-🏋 Workout Plan:
-- 3-5 days exercise per week
-- Cardio + Strength training
-- Beginner → start light
-
-🥗 Diet Plan:
-- High protein foods
-- Reduce junk food
-- Drink 2-3L water daily
-
-💡 Tips:
-- Sleep 7-8 hours
-- Stay consistent
-- Track progress weekly
-"""
-
-       
-
+        return f"Error: {str(e)}"
 
+# Gradio UI
 interface = gr.Interface(
     fn=fitness_coach,
     inputs=[
         gr.Number(label="Age"),
         gr.Number(label="Weight (kg)"),
         gr.Dropdown(["Weight Loss", "Muscle Gain", "Maintenance"], label="Goal"),
-        gr.Dropdown(["Beginner", "Intermediate", "Advanced"], label="Activity Level"),
+        gr.Dropdown(["Beginner", "Intermediate", "Advanced"], label="Activity Level")
     ],
     outputs="text",
     title="💪 Personalized Fitness Coach",