Update app.py

app.py

@@ -64,9 +64,28 @@ Generated json
         try:
             genai.configure(api_key=api_key)
             
-            # Try different model names in order of preference
+            # Try to list available models first
+            try:
+                models = genai.list_models()
+                available_models = [m.name for m in models if 'generateContent' in m.supported_generation_methods]
+                
+                # Print available models for debugging
+                print(f"Available models: {available_models}")
+                
+                if available_models:
+                    # Extract just the model name without the prefix
+                    model_name = available_models[0].split('/')[-1]
+                    self.model = genai.GenerativeModel(
+                        model_name=model_name,
+                        system_instruction=self.SYSTEM_PROMPT
+                    )
+                    return f"✅ Model initialized successfully! Using: {model_name}"
+            except Exception as list_error:
+                print(f"Error listing models: {list_error}")
+                
+            # If listing models fails, try these specific models
             model_names = [
-                'gemini-1.5-pro-latest',
+                'gemini-pro',
                 'gemini-1.5-pro',
                 'gemini-1.5-flash',
                 'gemini-1.0-pro'
@@ -80,22 +99,10 @@ Generated json
                     )
                     return f"✅ Model initialized successfully! Using: {model_name}"
                 except Exception as model_error:
+                    print(f"Error with model {model_name}: {model_error}")
                     continue
             
-            # If all models fail, list available models
-            models = genai.list_models()
-            available_models = [m.name for m in models if 'generateContent' in m.supported_generation_methods]
-            
-            if available_models:
-                # Use the first available model
-                model_name = available_models[0].replace('models/', '')
-                self.model = genai.GenerativeModel(
-                    model_name=model_name,
-                    system_instruction=self.SYSTEM_PROMPT
-                )
-                return f"✅ Model initialized successfully! Using: {model_name}"
-            else:
-                return "❌ No compatible models found"
+            return "❌ No compatible models found. Please check your API key and available models."
                 
         except Exception as e:
             return f"❌ Error initializing model: {str(e)}"
@@ -107,7 +114,7 @@ Generated json
         
         if not self.model:
             init_result = self.initialize_model(api_key)
-            if "Error" in init_result:
+            if "Error" in init_result or "❌" in init_result:
                 return None, None, init_result
         
         if not core_prompt.strip():