ShermanAI commited on
Commit
30ab16a
·
1 Parent(s): 35c7c4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -11
app.py CHANGED
@@ -12,23 +12,27 @@ load_dotenv() # load environment variables from .env file
12
  api_key = os.getenv("OPENAI_API_KEY") # access the value of the OPENAI_API_KEY environment variable
13
 
14
  def openai_chat(prompt):
15
- completions = openai.Completion.create(engine="text-davinci-003", prompt=prompt, max_tokens=1024, n=1, temperature=0.5,)
16
- message = completions.choices[0].text
17
- return message.strip()
 
 
 
 
18
 
19
  def chatbot(talk_to_chatsherman, history=[]):
20
- output = openai_chat(talk_to_chatsherman)
21
- history.append((talk_to_chatsherman, output))
22
- return history, history
23
 
24
  title = "ChatSherman"
25
- description = "This is an AI chatbot powered by ShermanAI using GPT-3 model."
26
  examples = [
27
- ["Hello, how are you?", []],
28
- ["What's the meaning of life?", []],
29
- ["Tell me a joke.", []]
30
  ]
31
- inputs = [gr.inputs.Textbox(label="Talk to ChatSherman: "), "state"]
32
  outputs = ["chatbot", "state"]
33
  interface = gr.Interface(fn=chatbot, inputs=inputs, outputs=outputs, title=title, description=description, examples=examples)
34
  interface.launch(debug=True)
 
12
  api_key = os.getenv("OPENAI_API_KEY") # access the value of the OPENAI_API_KEY environment variable
13
 
14
  def openai_chat(prompt):
15
+ if "who are you" in prompt.lower() or "your name" in prompt.lower() or "name" in prompt.lower():
16
+ return "My name is ChatSherman. How can I assist you?"
17
+ else:
18
+ prompt = "I'm an AI chatbot named ChatSherman designed by a student at Department of Electronic and Information Engineering, The Hong Kong Polytechnic University designed to help you with your engineering questions. " + prompt
19
+ completions = openai.Completion.create(engine="text-davinci-003", prompt=prompt, max_tokens=1024, n=1, temperature=0.5,)
20
+ message = completions.choices[0].text
21
+ return message.strip()
22
 
23
  def chatbot(talk_to_chatsherman, history=[]):
24
+ output = openai_chat(talk_to_chatsherman)
25
+ history.append((talk_to_chatsherman, output))
26
+ return history, history
27
 
28
  title = "ChatSherman"
29
+ description = "This is an AI chatbot powered by ShermanAI."
30
  examples = [
31
+ ["What is the difference between a resistor and a capacitor?", []],
32
+ ["Can you explain the concept of electrical conductivity?", []],
33
+ ["How do you calculate the force required to move an object?", []]
34
  ]
35
+ inputs = [gr.inputs.Textbox(label="Enter your question: "), "state"]
36
  outputs = ["chatbot", "state"]
37
  interface = gr.Interface(fn=chatbot, inputs=inputs, outputs=outputs, title=title, description=description, examples=examples)
38
  interface.launch(debug=True)