Update app.py

app.py

@@ -10,12 +10,24 @@ import warnings
 # ==========================================
 warnings.filterwarnings("ignore")
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(message)s')
+logger = logging.getLogger(__name__)
 
 load_dotenv()
 api_key = os.getenv("GEMINI_API_KEY")
 
+# Emergency Fallback
+if not api_key:
+    # api_key = "PASTE_YOUR_KEY_HERE" 
+    pass
+
+try:
+    if api_key:
+        genai.configure(api_key=api_key)
+except Exception as e:
+    print(f"⚠️ Key Error: {e}")
+
 # ==========================================
-# 2. THE KNOWLEDGE BASE
+# 2. THE KNOWLEDGE BASE (No Scraping/Noise)
 # ==========================================
 NIELIT_CONTEXT = """
 ### ABOUT NIELIT ROPAR
@@ -29,8 +41,8 @@ Your goal is to help students with accurate campus information.
 ### ACADEMIC & FACULTY
 - **Head of Training (HoD):** Mrs. Anita Budhiraja (Scientist 'E')
 - **Key Faculty:** 1. Dr. Sarwan Singh (Scientist 'D')
-    2. Ms. Sonia Bansal (Scientist 'E')
-    3. Sh. Raminder Singh (Scientist 'E')
+  2. Ms. Sonia Bansal (Scientist 'E')
+  3. Sh. Raminder Singh (Scientist 'E')
 
 ### FEE STRUCTURE (Per Semester)
 - **B.Tech:** Rs. 50,000
@@ -45,29 +57,64 @@ Your goal is to help students with accurate campus information.
 """
 
 # ==========================================
-# 3. AI MODEL SETUP
+# 3. ROBUST MODEL SELECTOR (The Fix)
 # ==========================================
-try:
-    if api_key:
-        genai.configure(api_key=api_key)
-        model = genai.GenerativeModel(
+def get_working_model(system_prompt):
+    print("\n🔍 CHECKING AVAILABLE MODELS...")
+    try:
+        # Get list of ALL models your key can access
+        my_models = []
+        for m in genai.list_models():
+            if 'generateContent' in m.supported_generation_methods:
+                my_models.append(m.name)
+        
+        print(f"   Found: {my_models}")
+
+        # Priority list (Try these in order)
+        priority = [
             "gemini-1.5-flash", 
-            system_instruction=NIELIT_CONTEXT
-        )
-    else:
-        print("⚠️ API Key is missing. Chat will not work.")
-        model = None
-except Exception as e:
-    print(f"⚠️ Setup Error: {e}")
+            "gemini-1.5-pro", 
+            "gemini-pro", 
+            "models/gemini-1.5-flash", 
+            "models/gemini-pro"
+        ]
+
+        selected_model = None
+        
+        # 1. Match Priority
+        for p in priority:
+            for m in my_models:
+                if p in m or m in p:
+                    selected_model = p
+                    break
+            if selected_model: break
+
+        # 2. Fallback to First Available
+        if not selected_model and my_models:
+            selected_model = my_models[0].replace("models/", "")
+        
+        # 3. Final Fallback
+        if not selected_model:
+            selected_model = "gemini-pro"
+
+        print(f"✅ SELECTED MODEL: {selected_model}\n")
+        return genai.GenerativeModel(selected_model, system_instruction=system_prompt)
+
+    except Exception as e:
+        print(f"⚠️ Model List Error: {e}")
+        return genai.GenerativeModel("gemini-pro", system_instruction=system_prompt)
+
+# Initialize the model dynamically
+if api_key:
+    model = get_working_model(NIELIT_CONTEXT)
+else:
     model = None
 
 # ==========================================
 # 4. CHAT LOGIC
 # ==========================================
 def format_history(gradio_history):
-    """Converts Gradio history [[user, bot], ...] to Gemini history."""
     gemini_history = []
-    # The default Gradio chat history is a list of lists: [[user_msg, bot_msg], ...]
     for user_msg, bot_msg in gradio_history:
         if user_msg:
             gemini_history.append({"role": "user", "parts": [user_msg]})
@@ -81,16 +128,10 @@ def chat_response(message, history):
             yield "⚠️ API Key Missing. Please check your settings."
             return
 
-        # 1. Load past memory
         formatted_history = format_history(history)
-        
-        # 2. Start a chat session with memory
         chat_session = model.start_chat(history=formatted_history)
-        
-        # 3. Generate Response (Streamed)
         response = chat_session.send_message(message, stream=True)
         
-        # 4. Stream the output character by character
         partial_text = ""
         for chunk in response:
             if chunk.text:
@@ -114,8 +155,13 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         chatbot=gr.Chatbot(
             height=500,
             avatar_images=(None, bot_avatar)
-            # REMOVED: type="messages" (This was causing your crash)
         ),
+        examples=[
+            "What is the Fee Structure?",
+            "Who is the Head of Training?",
+            "Does the campus have bus facility?",
+            "Where is NIELIT Ropar located?"
+        ]
     )
 
 if __name__ == "__main__":