Update app.py

app.py

@@ -19,18 +19,55 @@ custom_css = """
     border: 1 px solid white;
 }
 """
+
+def predict(prompt):
+    generation_config = {
+      "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-2.0-flash-exp",
+      generation_config=generation_config,
+    )
+
+    chat_session = model.start_chat(
+      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(f"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: \n {prompt}")
+    return response
+    
 def get_impressum_text(search_term):
     
     headers = {
         "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
     }
-    
+    vereine = []
     #search_results = google_search(search_term)
     url = f"https://www.google.com/search?q=mpressum {search_term}"
     response = requests.get(url, headers=headers)
     soup = BeautifulSoup(response.content, 'html.parser')
     impressum_div = soup.find('body')
-    return impressum_div.text
+    json_data = predict(impressum_div.text)
+    vereine.append(json_data)
+    return vereine
 
 def websearch(prompt):
     
@@ -56,38 +93,6 @@ def websearch(prompt):
         print("Kein div mit der Klasse 'MjjYud' gefunden.")
         return None
 
-def predict(prompt):
-    generation_config = {
-      "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-2.0-flash-exp",
-      generation_config=generation_config,
-    )
-
-    chat_session = model.start_chat(
-      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("INSERT_INPUT_HERE")
 
 
 # Create the Gradio interface