Update app.py

app.py

@@ -32,16 +32,6 @@ def get_impressum_text(search_term):
     impressum_div = soup.find('body')
     return impressum_div.text
 
-    if 'items' in search_results:
-        for item in search_results['items']:
-            link = item['link']
-            response = requests.get(link, timeout=5) # Timeout hinzugefügt für Fehlerbehandlung
-            response.raise_for_status() # Wirft eine Exception, wenn der Statuscode nicht 200 ist
-            soup = BeautifulSoup(response.content, 'html.parser')
-
-            impressum_div = soup.find('div', class_='MjjYud')
-            return impressum_div.text.strip()
-
 def websearch(prompt):
     
     url = f"https://www.google.com/search?q={prompt}"
@@ -67,29 +57,38 @@ def websearch(prompt):
         return None
 
 def predict(prompt):
-    # Create the model
     generation_config = {
-        "temperature": 0.3,
-        "top_p": 0.95,
-        "top_k": 40,
-        "max_output_tokens": 2048,
-        "response_mime_type": "text/plain",
+      "temperature": 0.4,
+      "top_p": 0.95,
+      "top_k": 40,
+      "max_output_tokens": 8192,
+      "response_mime_type": "text/plain",
     }
 
     model = genai.GenerativeModel(
-        #model_name="gemini-1.5-pro",
-        model_name="gemini-2.0-flash-exp",
-        generation_config=generation_config,
+      model_name="gemini-2.0-flash-exp",
+      generation_config=generation_config,
     )
 
     chat_session = model.start_chat(
-        history=[
-        ]
+      history=[
+        {
+          "role": "user",
+          "parts": [
+            "return a json object with the contact details. leave blank if information is not available. here is the json schema:\n\n{\n  \"organization\": \"\",\n  \"address\": \"\",\n  \"phone\": \"\",\n  \"email\": \"\",\n  \"website\": \"\"\n}\n\nyou can find the contact details here: \nImpressum – Aero-Club Bamberg e.V.\nAero-Club Bamberg\nhttps://aeroclub-bamberg.de › impressum\nAero-Club Bamberg e.V.. Zeppelinstraße 18. D-96052 Bamberg. Tel.: 0951 / 45 1 45. Fax: 0951 / 13 22 20. E-Mail: [email protected].\n",
+          ],
+        },
+        {
+          "role": "model",
+          "parts": [
+            "```json\n{\n  \"organization\": \"Aero-Club Bamberg e.V.\",\n  \"address\": \"Zeppelinstraße 18, D-96052 Bamberg\",\n  \"phone\": \"0951 / 45 1 45\",\n  \"email\": \"[email protected]\",\n  \"website\": \"https://aeroclub-bamberg.de\"\n}\n```\n",
+          ],
+        },
+      ]
     )
-    
-    response = chat_session.send_message(prompt)
-    #response = model.generate_content(contents=prompt, tools='google_search_retrieval')
-    return response.text
+
+    response = chat_session.send_message("INSERT_INPUT_HERE")
+
 
 # Create the Gradio interface
 with gr.Blocks(css=custom_css) as demo: