Update app.py

app.py

@@ -2,15 +2,21 @@ import gradio as gr
 import google.generativeai as genai
 import os
 
-# Configure API key
-genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+# 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)
+
+model = genai.GenerativeModel("gemini-pro")
 
-model = genai.GenerativeModel("gemini-1.5-flash")
 
 def fitness_coach(age, weight, goal, activity):
     try:
         prompt = f"""
-        You are a certified fitness coach and nutrition expert.
+        You are a certified fitness coach.
 
         Create a personalized fitness plan:
 
@@ -26,18 +32,20 @@ def fitness_coach(age, weight, goal, activity):
         """
 
         response = model.generate_content(prompt)
-        return response.text
+
+        return response.text if response.text else "No response generated"
 
     except Exception as e:
         return f"Error: {str(e)}"
 
+
 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",