Update app.py

Final fix - working model

app.py

@@ -5,61 +5,41 @@ import os
 # Configure API key
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-# Use correct model
-model = genai.GenerativeModel("gemini-pro")
+# Use working model
+model = genai.GenerativeModel("gemini-1.5-flash")
 
 def fitness_coach(age, weight, goal, activity):
     try:
         prompt = f"""
-You are a professional fitness coach.
+        You are a certified fitness coach.
 
-Generate a UNIQUE and personalized fitness plan based on the user details.
+        Create a personalized fitness plan:
 
-User Details:
-Age: {age}
-Weight: {weight} kg
-Goal: {goal}
-Activity Level: {activity}
+        Age: {age}
+        Weight: {weight}
+        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,
-            generation_config={
-                "temperature": 0.9,
-                "top_p": 1,
-                "top_k": 1,
-                "max_output_tokens": 500,
-            }
-        )
+        Include:
+        1. Workout Plan
+        2. Diet Plan
+        3. Tips
+        """
 
+        response = model.generate_content(prompt)
         return response.text
 
     except Exception as e:
         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",