Update app.py

app.py

@@ -11,6 +11,7 @@ from selenium.webdriver.common.keys import Keys
 import groq
 import os
 import time
+import requests
 
 options = webdriver.ChromeOptions()
 options.add_argument('--headless')
@@ -20,6 +21,31 @@ wd = webdriver.Chrome(options=options)
 api_key = os.getenv('groq')
 client = groq.Client(api_key=api_key)
 
+
+def impressum_suche(message):
+  """
+  Führt eine Anfrage an DuckDuckGo durch, um nach einem Impressum in Verbindung mit einer Nachricht zu suchen.
+
+  Args:
+    message: Der Suchbegriff, der zusammen mit "Impressum" verwendet wird.
+
+  Returns:
+    Den Inhalt der Antwort als Text, oder None im Fehlerfall.
+  """
+  url = f"https://duckduckgo.com/?q=impressum {message}"
+  try:
+    response = requests.get(url)
+    response.raise_for_status()  # Wirft eine Ausnahme für ungültige Statuscodes
+    return response.text
+  except requests.exceptions.RequestException as e:
+    print(f"Fehler beim Abrufen der URL: {e}")
+    return None
+
+# Beispielaufruf:
+if __name__ == "__main__":
+  suchbegriff = "Beispielunternehmen"
+  ergebnis = impressum_suche(suchbegriff)
+
 # Use Llama 3 70B powered by Groq for answering
 def update(prompt, ort):   
     try:        
@@ -39,16 +65,21 @@ def selenium(message):
     #url = 'https://www.spiegel.de'
     url = f"https://duckduckgo.com/?q=impressum {message}" 
     #<ol class="AmbQnf">
-    wd.get(url)
-    time.sleep(3)
-    element = wd.find_element(By.TAG_NAME, "body")
+
+    response = requests.get(url)
+    response.raise_for_status()  # Wirft eine Ausnahme für ungültige Statuscodes
+    return response.text
+
+    #wd.get(url)
+    #time.sleep(3)
+    #element = wd.find_element(By.TAG_NAME, "body")
     #return element.text     
     try:        
         completion = client.chat.completions.create(
             model="llama3-8b-8192",
             messages=[
                 {"role": "system", "content": "You are a helpful assistant."},
-                {"role": "user", "content": f"return json object with keys name and email. \n  name = {message}\n  value for email can be found here: {element.text} \n return json object only, no additional text or comments \n"}
+                {"role": "user", "content": f"return json object with keys name and email. \n  name = {message}\n  value for email can be found here: {response.text} \n return json object only, no additional text or comments \n"}
             ],
         )       
         return completion.choices[0].message.content