ShermanAI commited on
Commit
82e5470
·
1 Parent(s): 92db671

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -16
app.py CHANGED
@@ -12,25 +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
- topics = ["engineering", "science", "math", "IT", "history", "English"]
16
- prompt = f"Hello! My name is ChatSherman and I'm an AI chatbot designed to help you with your questions in {', '.join(topics)}. {prompt}"
17
- completions = openai.Completion.create(engine="text-davinci-003", prompt=prompt, max_tokens=1024, n=1, temperature=0.5,)
18
- message = completions.choices[0].text
19
- return message.strip()
 
 
20
 
21
- def chatbot(question):
22
- response = openai_chat(question)
23
- return response
 
24
 
25
- title = "ChatSherman - AI Chatbot"
26
- description = "This is an AI chatbot powered by ShermanAI"
27
  examples = [
28
- ["What is the Pythagorean theorem?"],
29
- ["Can you explain the concept of electrical conductivity?"],
30
- ["Who was the first president of the United States?"],
31
  ]
32
- inputs = [gr.inputs.Textbox(label="Enter your question: ")]
33
- outputs = gr.outputs.Textbox(label="ChatSherman's Response:")
34
-
35
  interface = gr.Interface(fn=chatbot, inputs=inputs, outputs=outputs, title=title, description=description, examples=examples)
36
  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 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)