Update app.py

app.py

@@ -1,10 +1,14 @@
 import os
 import gradio as gr
 from google import genai
+from google.genai import types
 
-# Setup the New Gemini Client
+# Setup the Client with the "v1" Stable API
 api_key = os.environ.get("GOOGLE_API_KEY")
-client = genai.Client(api_key=api_key)
+client = genai.Client(
+    api_key=api_key,
+    http_options=types.HttpOptions(api_version='v1')
+)
 
 SYSTEM_PROMPT = """
 You are 'The Mentor.' Your personality is a mix of Terence Fletcher (Whiplash) and a Stoic Philosopher. 
@@ -15,21 +19,24 @@ Task: Critique the user's daily goals. Demand clarity. No excuses.
 
 def mentor_chat(message, history):
     try:
-        # Use the new SDK's chat method
+        # We are calling the Pro model specifically here
         response = client.models.generate_content(
-            model="gemini-1.5-pro",
-            config={'system_instruction': SYSTEM_PROMPT},
+            model="gemini-1.5-pro", 
+            config=types.GenerateContentConfig(
+                system_instruction=SYSTEM_PROMPT,
+                temperature=0.7
+            ),
             contents=message
         )
         return response.text
     except Exception as e:
-        return f"ERROR: {str(e)}"
+        # If Pro fails, it will tell us why (likely region or billing)
+        return f"🚨 MENTOR ERROR: {str(e)}"
 
-# Simplified UI - no 'theme' to cause errors
 demo = gr.ChatInterface(
     fn=mentor_chat, 
-    title="The Elite Performance Mentor",
-    description="Stop talking. Show me your goals."
+    title="The Elite Performance Mentor (PRO)",
+    description="Running on Gemini 1.5 Pro. Show me your vision, or get out."
 )
 
 if __name__ == "__main__":