masadonline commited on
Commit
ab4f2f9
·
verified ·
1 Parent(s): e98797d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
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 openai
8
  from dotenv import load_dotenv
9
 
10
  load_dotenv()
11
  GROQ_API_KEY = os.getenv("GROQ_API_KEY")
12
- openai.api_key = GROQ_API_KEY
13
- openai.api_base = "https://api.groq.com/openai/v1"
 
 
 
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 = openai.ChatCompletion.create(
64
  model="llama3-8b-8192",
65
  messages=[
66
  {"role": "system", "content": system_prompt},
67
  {"role": "user", "content": query}
68
  ]
69
  )
70
- answer = response['choices'][0]['message']['content']
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: