Update app.py

app.py

@@ -1,14 +1,19 @@
 import os
 import gradio as gr
 from google import genai
+from google.genai import types
 from google.genai.types import GenerateContentConfig, GoogleSearch, Tool
 
-# -------- Gemini Setup --------
+# -----------------------
+# Gemini Setup
+# -----------------------
 API_KEY = os.getenv("Gemini_API_Key")
 client = genai.Client(api_key=API_KEY)
 MODEL_ID = "gemini-2.0-flash"
 
-# -------- Custom CSS + JS --------
+# -----------------------
+# Custom CSS + JS
+# -----------------------
 custom_css = """
 #search-btn {
     background-color: #4f46e5 !important;
@@ -18,7 +23,6 @@ custom_css = """
     padding: 10px 18px;
     transition: opacity 0.6s ease, transform 0.2s ease;
 }
-
 #search-btn:active {
     opacity: 0.5;
     transform: scale(0.97);
@@ -32,32 +36,52 @@ custom_js = """
 }
 """
 
-# -------- AI Function --------
+# -----------------------
+# AI Function (FIXED)
+# -----------------------
 def google_search_query(question):
     try:
+        if not question or question.strip() == "":
+            return "Please enter a question.", ""
+
         google_search_tool = Tool(google_search=GoogleSearch())
 
         response = client.models.generate_content(
             model=MODEL_ID,
             contents=[
-    {
-        "role": "user",
-        "parts": [{"text": question}]
-    }
-]
-,
-            config=GenerateContentConfig(tools=[google_search_tool]),
+                types.Content(
+                    role="user",
+                    parts=[types.Part.from_text(question)]
+                )
+            ],
+            config=GenerateContentConfig(
+                tools=[google_search_tool]
+            ),
         )
 
-        ai_response = response.text
-        search_results = response.candidates[0].grounding_metadata.search_entry_point.rendered_content
+        ai_response = response.text or "No AI response generated."
+
+        search_results = ""
+        if (
+            response.candidates
+            and response.candidates[0].grounding_metadata
+            and response.candidates[0].grounding_metadata.search_entry_point
+        ):
+            search_results = (
+                response.candidates[0]
+                .grounding_metadata
+                .search_entry_point
+                .rendered_content
+            )
 
         return ai_response, search_results
 
     except Exception as e:
         return f"Error: {str(e)}", ""
 
-# -------- Gradio App --------
+# -----------------------
+# Gradio App
+# -----------------------
 with gr.Blocks(css=custom_css) as app:
     gr.Markdown("## 🔍 Google Search with Gemini AI")
 
@@ -75,5 +99,8 @@ with gr.Blocks(css=custom_css) as app:
         js=custom_js
     )
 
+# -----------------------
+# Launch
+# -----------------------
 if __name__ == "__main__":
     app.launch(share=True)