Update app.py

app.py

@@ -6,12 +6,34 @@ from gradio_client import Client
 from selenium.webdriver.support.wait import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.webdriver.chrome.options import Options
+from gradio_client import Client
+import speech_recognition as sr
+import groq
+import os
 import time
 
 options = webdriver.ChromeOptions()
 options.add_argument('--headless')
 wd = webdriver.Chrome(options=options)
 
+#get your api-key @groq.com. its free!
+api_key = os.getenv('groq')
+client = groq.Client(api_key=api_key)
+
+# Use Llama 3 70B powered by Groq for answering
+def update(prompt, ort):   
+    try:        
+        completion = client.chat.completions.create(
+            model="Mixtral-8x7b-32768",
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant."},
+                {"role": "user", "content": f". gefragt sind die nächsten 3 zugverbindungen von bad kissingen nach {ort} du findest die antwort im kontext. liefer als antwort ein 2 spaltige tabelle. linke spalte: abfahrtszeit, fahrtdauer, ankunftszeit.rechte spalte: abfahrtsort,leer,zielort. formatiere die tabelle in markdown\n kontext: \n {prompt}"}
+            ],
+        )       
+        return completion.choices[0].message.content
+    except Exception as e:
+        return f"Error in response generation: {str(e)}"
+
 def selenium(message):
 
     try:       
@@ -23,7 +45,8 @@ def selenium(message):
         element = wd.find_element(By.TAG_NAME, "body")
         #texts = [element.text for element in elements]
         time.sleep(1)
-        return element.text
+        results = update(message,element.text)
+        return results
 
     except WebDriverException as e:
         return "fehler"