Update app.py

app.py

@@ -42,6 +42,19 @@ def selenium(message):
     wd.get(url)
     time.sleep(1)
     element = wd.find_element(By.TAG_NAME, "body")
+      
+    try:        
+        completion = client.chat.completions.create(
+            model="llama3-70b-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  \n"}
+            ],
+        )       
+        return completion.choices[0].message.content
+    except Exception as e:
+        return f"Error in response generation: {str(e)}"
+   
     return element.text
     #text_input = wd.find_element(By.XPATH, '/html/body/div[1]/div/div[2]/div[2]/div/div/div[2]/div[2]/div/div/div[1]/textarea')