Update app.py

app.py

@@ -10,47 +10,50 @@ MODEL_ID = "gemini-2.0-flash-exp"  # Replace with your desired model ID
 
 def google_search_query(question, use_web_search):
     try:
+        # Generate the AI response without web search initially
+        response = client.models.generate_content(
+            model=MODEL_ID,
+            contents=question,
+        )
+        ai_response = response.text  # AI response as plain text
+
         if use_web_search:
+            # If user selects web search, redefine the tool and generate response with web search
             google_search_tool = Tool(google_search=GoogleSearch())
             response = client.models.generate_content(
                 model=MODEL_ID,
                 contents=question,
                 config=GenerateContentConfig(tools=[google_search_tool]),
             )
-            ai_response = response.text  # AI response as plain text
-
-            # Check if search results are available and extract them
-            if hasattr(response.candidates[0].grounding_metadata, 'search_entry_point'):
-                search_results = response.candidates[0].grounding_metadata.search_entry_point
-                links_html = "\n".join([f"<a href='{result.url}' target='_blank'>{result.title}</a>" for result in search_results.results])
-            else:
-                links_html = "No search results found."
-
-            return ai_response, links_html
+            search_results = response.candidates[0].grounding_metadata.search_entry_point.rendered_content
         else:
-            # Return response without web search
-            response = client.models.generate_content(
-                model=MODEL_ID,
-                contents=question,
-            )
-            return response.text, "Web search not used."
+            search_results = "Web search not used."
+
+        return ai_response, search_results
     except Exception as e:
-        return f"Error: {str(e)}", "No results found."
+        return f"Error: {str(e)}", ""
 
+# Gradio Interface using Chatbot and conditional web search
 with gr.Blocks() as app:
     with gr.Row():
-        question_input = gr.Textbox(label="Ask a Question", lines=2)
-        web_search_checkbox = gr.Checkbox(label="Enable Web Search", value=False)
+        question_input = gr.Textbox(lines=2, label="Ask a Question")
+        web_search_checkbox = gr.Checkbox(label="Enhance with Web Search", value=False)
     with gr.Row():
         submit_button = gr.Button("Submit")
-    with gr.Row():
-        ai_response_output = gr.Textbox(label="AI Response")
-        search_results_output = gr.HTML(label="Search Results")
-
+    chatbot = gr.Chatbot(label="Chatbot Responses")
+    
+    def update_chatbot(question, use_web_search, chat_log):
+        ai_response, search_results = google_search_query(question, use_web_search)
+        chat_log.append(("You", question))
+        chat_log.append(("AI", ai_response))
+        if use_web_search:
+            chat_log.append(("AI + Web Search", search_results))
+        return chat_log
+    
     submit_button.click(
-        google_search_query,
-        inputs=[question_input, web_search_checkbox],
-        outputs=[ai_response_output, search_results_output]
+        fn=update_chatbot,
+        inputs=[question_input, web_search_checkbox, chatbot],
+        outputs=[chatbot]
     )
 
-app.launch(share=True)
+app.launch()