Update app.py

app.py

@@ -49,21 +49,25 @@ custom_js = """
 # AI Function (BULLETPROOF)
 # =========================================================
 def google_search_query(question):
-    try:
-        # Force conversion to string safely
-        if question is None:
-            question = ""
-        else:
-            question = str(question)
 
-        question = question.strip()
+    # ---------- 1️⃣ Normalize safely ----------
+    if question is None:
+        question = ""
+    else:
+        question = str(question)
+
+    question = question.strip()
 
-        # Only block if truly empty
-        if len(question) == 0:
-            return "Please type a question above 👆", ""
+    if question == "":
+        return "Please type a question above 👆", ""
 
-        google_search_tool = Tool(google_search=GoogleSearch())
+    try:
+        # ---------- 2️⃣ Define Search Tool ----------
+        google_search_tool = Tool(
+            google_search=GoogleSearch()
+        )
 
+        # ---------- 3️⃣ Generate Response ----------
         response = client.models.generate_content(
             model=MODEL_ID,
             contents=[
@@ -77,17 +81,35 @@ def google_search_query(question):
             ),
         )
 
-        ai_response = response.text or "No AI response generated."
+        # ---------- 4️⃣ Extract AI Text Safely ----------
+        ai_response = ""
 
-        # Safe grounding extraction
+        if hasattr(response, "text") and response.text:
+            ai_response = response.text
+        elif response.candidates:
+            try:
+                ai_response = response.candidates[0].content.parts[0].text
+            except Exception:
+                ai_response = "No AI response generated."
+        else:
+            ai_response = "No AI response generated."
+
+        # ---------- 5️⃣ Extract Search Grounding Safely ----------
         search_results = ""
+
         try:
-            search_results = (
-                response.candidates[0]
-                .grounding_metadata
-                .search_entry_point
-                .rendered_content
-            )
+            candidate = response.candidates[0]
+            if (
+                hasattr(candidate, "grounding_metadata")
+                and candidate.grounding_metadata
+                and candidate.grounding_metadata.search_entry_point
+            ):
+                search_results = (
+                    candidate
+                    .grounding_metadata
+                    .search_entry_point
+                    .rendered_content
+                )
         except Exception:
             search_results = ""