Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -4,13 +4,16 @@ from PyPDF2 import PdfReader
|
|
4 |
import docx
|
5 |
import pandas as pd
|
6 |
from bs4 import BeautifulSoup
|
7 |
-
import
|
8 |
from dotenv import load_dotenv
|
9 |
|
10 |
load_dotenv()
|
11 |
GROQ_API_KEY = os.getenv("GROQ_API_KEY")
|
12 |
-
|
13 |
-
|
|
|
|
|
|
|
14 |
|
15 |
st.set_page_config(page_title="ToyShop Order Assistant", layout="wide")
|
16 |
st.title("🧸 Online Toy Shop - Order Status Assistant")
|
@@ -60,14 +63,14 @@ if query and combined_text:
|
|
60 |
"Answer customer queries based on the following order information:\n\n"
|
61 |
+ combined_text
|
62 |
)
|
63 |
-
response =
|
64 |
model="llama3-8b-8192",
|
65 |
messages=[
|
66 |
{"role": "system", "content": system_prompt},
|
67 |
{"role": "user", "content": query}
|
68 |
]
|
69 |
)
|
70 |
-
answer = response
|
71 |
st.success("Answer:")
|
72 |
st.write(answer)
|
73 |
except Exception as e:
|
|
|
4 |
import docx
|
5 |
import pandas as pd
|
6 |
from bs4 import BeautifulSoup
|
7 |
+
from openai import OpenAI
|
8 |
from dotenv import load_dotenv
|
9 |
|
10 |
load_dotenv()
|
11 |
GROQ_API_KEY = os.getenv("GROQ_API_KEY")
|
12 |
+
|
13 |
+
client = OpenAI(
|
14 |
+
api_key=GROQ_API_KEY,
|
15 |
+
base_url="https://api.groq.com/openai/v1" # required for Groq
|
16 |
+
)
|
17 |
|
18 |
st.set_page_config(page_title="ToyShop Order Assistant", layout="wide")
|
19 |
st.title("🧸 Online Toy Shop - Order Status Assistant")
|
|
|
63 |
"Answer customer queries based on the following order information:\n\n"
|
64 |
+ combined_text
|
65 |
)
|
66 |
+
response = client.chat.completions.create(
|
67 |
model="llama3-8b-8192",
|
68 |
messages=[
|
69 |
{"role": "system", "content": system_prompt},
|
70 |
{"role": "user", "content": query}
|
71 |
]
|
72 |
)
|
73 |
+
answer = response.choices[0].message.content
|
74 |
st.success("Answer:")
|
75 |
st.write(answer)
|
76 |
except Exception as e:
|