.

app.py

@@ -11,7 +11,9 @@ def chat_with_ai(prompt):
         return "Error: API key missing. Set GEMINI_API_KEY in Hugging Face secrets."
     try:
         genai.configure(api_key=api_key)
-        model = genai.GenerativeModel("gemini-pro")
+        available_models = [m.name for m in genai.list_models()]
+        model_name = "gemini-pro" if "gemini-pro" in available_models else available_models[0]
+        model = genai.GenerativeModel(model_name)
         response = model.generate_content(prompt)
         return response.text
     except Exception as e: