Update app.py
Browse files
app.py
CHANGED
@@ -47,13 +47,13 @@ def selenium(message):
|
|
47 |
time.sleep(3)
|
48 |
element = wd.find_element(By.TAG_NAME, "body")
|
49 |
#time.sleep(3)
|
50 |
-
return element.text
|
51 |
try:
|
52 |
completion = client.chat.completions.create(
|
53 |
model="llama3-8b-8192",
|
54 |
messages=[
|
55 |
{"role": "system", "content": "You are a helpful assistant."},
|
56 |
-
{"role": "user", "content": f"return json object with keys name and email. \n name = {message}\n value for email can be found here: {
|
57 |
],
|
58 |
)
|
59 |
return completion.choices[0].message.content
|
|
|
47 |
time.sleep(3)
|
48 |
element = wd.find_element(By.TAG_NAME, "body")
|
49 |
#time.sleep(3)
|
50 |
+
#return element.text
|
51 |
try:
|
52 |
completion = client.chat.completions.create(
|
53 |
model="llama3-8b-8192",
|
54 |
messages=[
|
55 |
{"role": "system", "content": "You are a helpful assistant."},
|
56 |
+
{"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"}
|
57 |
],
|
58 |
)
|
59 |
return completion.choices[0].message.content
|