Update app.py

app.py

@@ -2,6 +2,24 @@ import gradio as gr
 from googlesearch import search
 import requests
 from bs4 import BeautifulSoup
+from gradio_client import Client
+import groq
+
+api_key = os.getenv('groq')
+client = groq.Client(api_key=api_key)
+
+def llm(message):   
+    try:        
+        completion = client.chat.completions.create(
+            model="Mixtral-8x7b-32768",
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant."},
+                {"role": "user", "content": f"{message}. antworte immer auf deutsch"}
+            ],
+        )       
+        return completion.choices[0].message.content
+    except Exception as e:
+        return f"Error in response generation: {str(e)}"
 
 def google_search(query):       
     headers = {
@@ -10,20 +28,14 @@ def google_search(query):
     # Führt die Suche durch und erhält das erste Ergebnis
     for result in search(query, num_results=1):
         url = result
-        #return url
         break
         
-    # Holt den Inhalt der Webseite
     response = requests.get(url)
     soup = BeautifulSoup(response.text, 'html.parser')  
     #first_div = soup.find('div', class_='MjjYud')
     first_div = soup.find('body')
     
-    if first_div:
-        return first_div.text.strip()
-    else:
-        print("Kein div mit der Klasse 'MjjYud' gefunden.")
-        return None
+    return first_div.text.strip()
 
 demo = gr.Interface(
     fn=google_search,