Update app.py

app.py

@@ -85,20 +85,40 @@ GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
 
 if GEMINI_API_KEY:
     genai.configure(api_key=GEMINI_API_KEY)
-    # Use the correct model name for the stable API
+    
+    # List available models to help debug
     try:
-        # Try gemini-pro first (most stable)
-        gemini_model = genai.GenerativeModel('gemini-pro')
-        print("✓ Gemini API initialized successfully with gemini-pro!")
+        print("Available Gemini models:")
+        for m in genai.list_models():
+            if 'generateContent' in m.supported_generation_methods:
+                print(f"  - {m.name}")
     except Exception as e:
-        print(f"Failed to load gemini-pro: {e}")
+        print(f"Could not list models: {e}")
+    
+    # Try different model names in order of preference
+    model_names_to_try = [
+        'models/gemini-1.5-flash',
+        'models/gemini-1.5-pro',
+        'models/gemini-pro',
+        'gemini-1.5-flash',
+        'gemini-1.5-pro',
+        'gemini-pro'
+    ]
+    
+    gemini_model = None
+    for model_name in model_names_to_try:
         try:
-            # Fallback to gemini-1.5-flash-latest
-            gemini_model = genai.GenerativeModel('gemini-1.5-flash-latest')
-            print("✓ Gemini API initialized successfully with gemini-1.5-flash-latest!")
-        except Exception as e2:
-            print(f"Failed to load gemini-1.5-flash-latest: {e2}")
-            gemini_model = None
+            gemini_model = genai.GenerativeModel(model_name)
+            # Test the model with a simple query
+            test_response = gemini_model.generate_content("Hi")
+            print(f"✓ Gemini API initialized successfully with {model_name}!")
+            break
+        except Exception as e:
+            print(f"Failed to load {model_name}: {e}")
+            continue
+    
+    if gemini_model is None:
+        print("⚠️ Warning: Could not initialize any Gemini model")
 else:
     gemini_model = None
     print("⚠️ Warning: GEMINI_API_KEY not found in environment variables")