Update app.py

app.py

@@ -41,7 +41,15 @@ def run_sql(sql):
 
 # 3. AGENTIC LOGIC
 def sql_agent(user_prompt):
-    model = genai.GenerativeModel('gemini-1.5-flash-latest')
+    # Force the model to use the most stable string
+# Try 'gemini-1.5-flash' first, but if it fails, 'gemini-pro' is the universal fallback
+try:
+    model = genai.GenerativeModel('gemini-1.5-flash')
+    # Test a small call to see if it's alive
+    model.generate_content("test") 
+except:
+    # Fallback to the older, most stable naming convention
+    model = genai.GenerativeModel('gemini-pro')
     schema = get_schema()
     
     # ATTEMPT 1