Update app.py

app.py

@@ -21,6 +21,34 @@ custom_css = """
 }
 """
 
+def google_search(search_term, **kwargs):
+    service = build("customsearch", "v1", developerKey=api_key) # Ersetze YOUR_GOOGLE_API_KEY mit deinem API-Key
+    res = service.cse().list(q=search_term, cx="77f1602c0ff764edb", **kwargs).execute() # Ersetze YOUR_CUSTOM_SEARCH_ENGINE_ID mit deiner Custom Search Engine ID
+    return res
+
+def get_impressum_text(search_term):
+    try:
+      search_results = google_search(search_term)
+
+      if 'items' in search_results:
+          for item in search_results['items']:
+              link = item['link']
+              try:
+                  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')
+
+                  if impressum_div:
+                      return impressum_div.text.strip()
+
+              except requests.exceptions.RequestException as e:
+                  print(f"Fehler beim Abrufen der URL {link}: {e}")
+                  continue # Weiter mit dem nächsten Suchergebnis
+
+      return None  # Kein Impressum gefunden
+
 
 def websearch(prompt):
     
@@ -82,7 +110,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=get_impressum_text, inputs=ort_input, outputs=details_output)   
 
 # Launch the Gradio application
 demo.launch()