Update app.py

app.py

@@ -10,10 +10,25 @@ if not api_key:
 else:
     genai.configure(api_key=api_key)
 
+    # Get available models
+    try:
+        available_models = [model.name for model in genai.list_models()]
+        if "gemini-pro" not in available_models:
+            st.warning(f"'gemini-pro' not found. Using 'gemini-pro-latest' instead.")
+            model_name = "gemini-pro-latest"
+        else:
+            model_name = "gemini-pro"
+    except Exception as e:
+        st.error(f"Error fetching available models: {e}")
+        model_name = "gemini-pro-latest"
+
     def gemini_generate(prompt):
-        """Generate content using the Gemini model with default settings."""
-        response = genai.GenerativeModel("gemini-pro").generate_content(prompt)
-        return response.text
+        """Generate content using the Gemini model with error handling."""
+        try:
+            response = genai.GenerativeModel(model_name).generate_content(prompt)
+            return response.text
+        except Exception as e:
+            return f"Error generating response: {e}"
 
     class GeminiAgent:
         """A simple agent wrapper that uses the Gemini model for generating responses."""
@@ -97,7 +112,7 @@ else:
         )
         return combined_response
 
-    st.title(" EduConnect Underserved Institutions Chatbot")
+    st.title("EduConnect Underserved Institutions Chatbot")
     st.write(
         "Ask any question related to schools, colleges, or universities in underserved regions. "
         "The chatbot will analyze the challenges, propose actionable solutions, and provide examples of institutions when relevant."