Update app.py

app.py

@@ -4,16 +4,16 @@ from google import genai
 from google.genai import types
 from google.genai.types import GenerateContentConfig, GoogleSearch, Tool
 
-# -----------------------
+# =========================================================
 # Gemini Setup
-# -----------------------
-API_KEY = os.getenv("Gemini_API_Key")
+# =========================================================
+API_KEY = os.getenv("Gemini_API_Key")  # Set this in HF Secrets
 client = genai.Client(api_key=API_KEY)
 MODEL_ID = "gemini-2.0-flash"
 
-# -----------------------
+# =========================================================
 # Custom CSS + JS
-# -----------------------
+# =========================================================
 custom_css = """
 #search-btn {
     background-color: #4f46e5 !important;
@@ -36,16 +36,19 @@ custom_js = """
 }
 """
 
-# -----------------------
+# =========================================================
 # AI Function
-# -----------------------
+# =========================================================
 def google_search_query(question):
     try:
+        # Guard: empty input
         if not question or question.strip() == "":
             return "Please enter a question.", ""
 
+        # Google Search tool
         google_search_tool = Tool(google_search=GoogleSearch())
 
+        # Gemini request (STRICT schema)
         response = client.models.generate_content(
             model=MODEL_ID,
             contents=[
@@ -59,8 +62,10 @@ def google_search_query(question):
             ),
         )
 
+        # AI text response
         ai_response = response.text or "No AI response generated."
 
+        # Grounded search results (safe access)
         search_results = ""
         if (
             response.candidates
@@ -79,28 +84,41 @@ def google_search_query(question):
     except Exception as e:
         return f"Error: {str(e)}", ""
 
-# -----------------------
+# =========================================================
 # Gradio App
-# -----------------------
+# =========================================================
 with gr.Blocks(css=custom_css) as app:
     gr.Markdown("## 🔍 Google Search with Gemini AI")
 
-    question = gr.Textbox(lines=2, label="Ask a Question")
+    question = gr.Textbox(
+        lines=2,
+        label="Ask a Question",
+        placeholder="e.g. Types of machine learning"
+    )
 
     search_btn = gr.Button("🔍 Search", elem_id="search-btn")
 
     ai_output = gr.Textbox(label="AI Response")
     search_output = gr.HTML(label="Search Results")
 
+    # Button click
     search_btn.click(
         fn=google_search_query,
-        inputs=[question],              # ✅ FIXED HERE
+        inputs=[question],
         outputs=[ai_output, search_output],
         js=custom_js
     )
 
-# -----------------------
+    # Press Enter to submit (UX improvement)
+    question.submit(
+        fn=google_search_query,
+        inputs=[question],
+        outputs=[ai_output, search_output],
+    )
+
+# =========================================================
 # Launch
-# -----------------------
+# =========================================================
 if __name__ == "__main__":
     app.launch(share=True)
+