Update app.py

app.py

@@ -4,6 +4,7 @@ import os
 import json
 import google.generativeai as genai
 from bs4 import BeautifulSoup
+from google.ai.generativelanguage_v1beta.types import content
 #from groq import Groq
 # Load environment variables
 genai.configure(api_key=os.environ["geminiapikey"])
@@ -21,6 +22,49 @@ custom_css = """
 }
 """
 
+generation_config = {
+  "temperature": 0.3,
+  "top_p": 0.95,
+  "top_k": 64,
+  "max_output_tokens": 8192,
+  "response_mime_type": "text/plain",
+}
+
+def ground_search(prompt):
+    model = genai.GenerativeModel(
+      model_name="gemini-2.0-pro-exp-02-05",
+      generation_config=generation_config,
+      tools = [
+        genai.protos.Tool(
+          google_search = genai.protos.Tool.GoogleSearch(),
+        ),
+      ],
+    )
+
+    chat_session = model.start_chat(
+      history=[
+        {
+          "role": "user",
+          "parts": [
+            "",
+          ],
+        },
+        {
+          "role": "model",
+          "parts": [
+            "",
+          ],
+        },
+      ]
+    )
+
+    response = chat_session.send_message(f"{prompt}")
+
+    #print(response.text)
+    return response.text
+
+
+
 #api_key = os.getenv('groq')
 google_api_key = os.getenv('google_search')
 API_URL = "https://blavken-flowiseblav.hf.space/api/v1/prediction/fbc118dc-ec00-4b59-acff-600648958be3"
@@ -97,7 +141,7 @@ with gr.Blocks(css=custom_css) as demo:
         button = gr.Button("Senden")    
 
     # Connect the button to the function
-    button.click(fn=websearch, inputs=ort_input, outputs=details_output)   
+    button.click(fn=ground_search, inputs=ort_input, outputs=details_output)   
 
 # Launch the Gradio application
 demo.launch()