Update app.py

app.py

@@ -42,6 +42,38 @@ When answering questions:
 5. Be encouraging and supportive to prospective students
 """
 
+def get_available_model_name():
+    """
+    Dynamically finds a working Gemini model from the user's account.
+    """
+    try:
+        available_models = []
+        for m in genai.list_models():
+            if 'generateContent' in m.supported_generation_methods:
+                available_models.append(m.name)
+        
+        if not available_models:
+            return None
+
+        # Priority list: Try to find these specific models first
+        preferred_order = [
+            "models/gemini-1.5-flash",
+            "models/gemini-1.5-pro",
+            "models/gemini-pro",
+            "models/gemini-1.0-pro"
+        ]
+        
+        # Check if any preferred model is available
+        for preferred in preferred_order:
+            if preferred in available_models:
+                return preferred
+
+        # If none of the preferred ones exist, use the first available
+        return available_models[0]
+
+    except Exception as e:
+        return None
+
 def get_chatbot_response(message, history):
     """Generate response using Gemini API"""
     
@@ -49,41 +81,14 @@ def get_chatbot_response(message, history):
         return "⚠️ Please set the GEMINI_API_KEY environment variable in Hugging Face Spaces settings."
     
     try:
-        # Try different model names in order of preference
-        model_names = [
-            'gemini-1.5-flash-latest',
-            'gemini-1.5-flash',
-            'gemini-1.5-pro-latest',
-            'gemini-1.5-pro',
-            'gemini-pro'
-        ]
-        
-        model = None
-        last_error = None
+        # Get the best available model
+        model_name = get_available_model_name()
         
-        # Try to initialize with available models
-        for model_name in model_names:
-            try:
-                model = genai.GenerativeModel(model_name)
-                # Test if model works with a simple generation
-                test_response = model.generate_content("Hi")
-                # If successful, break the loop
-                break
-            except Exception as e:
-                last_error = e
-                continue
+        if not model_name:
+            return "❌ No available Gemini models found for your API key.\n\n🔑 Please verify your API key at: https://aistudio.google.com/app/apikey\n\nMake sure you have access to Gemini models."
         
-        if model is None:
-            # If no model worked, try to list available models
-            try:
-                available_models = []
-                for m in genai.list_models():
-                    if 'generateContent' in m.supported_generation_methods:
-                        available_models.append(m.name)
-                
-                return f"❌ Could not initialize any Gemini model.\n\nAvailable models: {', '.join(available_models)}\n\nLast error: {str(last_error)}\n\nPlease check your API key at https://makersuite.google.com/app/apikey"
-            except:
-                return f"❌ Error: {str(last_error)}\n\nPlease ensure your GEMINI_API_KEY is valid. Get a new key at: https://aistudio.google.com/app/apikey"
+        # Initialize the model
+        model = genai.GenerativeModel(model_name)
         
         # Build conversation history
         chat_history = []
@@ -106,7 +111,7 @@ def get_chatbot_response(message, history):
         error_msg = str(e)
         if "API_KEY_INVALID" in error_msg or "invalid" in error_msg.lower():
             return f"❌ Invalid API Key Error: {error_msg}\n\n🔑 Please get a valid API key from: https://aistudio.google.com/app/apikey\n\nThen set it in Hugging Face Spaces: Settings → Repository secrets → GEMINI_API_KEY"
-        return f"❌ Error: {error_msg}\n\nIf the error persists, please verify your API key is correct."
+        return f"❌ Error: {error_msg}\n\nModel used: {model_name if 'model_name' in locals() else 'unknown'}\n\nIf the error persists, please verify your API key is correct."
 
 # Custom CSS for NIELIT branding
 custom_css = """