Update app.py

app.py

@@ -1,38 +1,39 @@
 import openai
 import gradio as gr
 
-model_id = "gpt-3.5-turbo"
-conversation = []
+class chatgpt:
 
-# Getting responses using the OpenAI API
-def answer_chatgpt(api_key, message, history):
-  history = history or []
-  prompt = f"{message}"
-  conversation.append({"role": "user", "content": f"{prompt}"})
-  # OPENAI API KEY
-  openai.api_key = api_key
-  response = openai.ChatCompletion.create(
-  model=model_id,
-  messages=conversation
-  )
-  # Displaying the answer on the screen
-  answer = response["choices"][0]["message"]["content"]
-  history.append((message, answer))
-  conversation.append({'role': response.choices[0].message.role, 'content': response.choices[0].message.content})
-  return history, history
+      def __init__(self):
+          self.model_id = "gpt-3.5-turbo"
+          self.conversation = []
 
+      # Getting responses using the OpenAI API
+      def answer_chatgpt(self, api_key, message, history):
+          self.history = history or []
+          prompt = f"{message}"
+          self.conversation.append({"role": "user", "content": f"{prompt}"})
+          # OPENAI API KEY
+          openai.api_key = api_key
+          response = openai.ChatCompletion.create(
+              model=self.model_id,
+              messages=self.conversation
+          )
+          # Displaying the answer on the screen
+          answer = response["choices"][0]["message"]["content"]
+          self.history.append((message, answer))
+          self.conversation.append({'role': response.choices[0].message.role, 'content': response.choices[0].message.content})
+          return self.history, self.history
+
+      def Clean(self):
+          self.history.clear()
+          self.conversation.clear()
 
-def Clean():
-    global history
-    global conversation
-    history = []
-    conversation = []
-      
 # User input
 block = gr.Blocks()
+chatgpt = chatgpt()
 
 with block:
-    gr.Markdown("""<h1><center>🤖 ChatGPT-Assistant 🐍</center></h1>
+    gr.Markdown("""<h1><center>🤖 ChatGPT-Assistant ðŸ</center></h1>
                    <p><center>ChatGPT-Assistant is a chatbot that uses the gpt-3.5-turbo model</center></p>
     """)
     api_key = gr.Textbox(type="password", label="Enter your OpenAI API key here")
@@ -40,9 +41,9 @@ with block:
     message = gr.Textbox(label="Message")
     state = gr.State()
     submit = gr.Button("Send")
-    submit.click(answer_chatgpt, inputs=[api_key, message, state], outputs=[chatbot, state])
+    submit.click(chatgpt.answer_chatgpt, inputs=[api_key, message, state], outputs=[chatbot, state])
     clean = gr.Button("Clean")
-    clean.click(Clean)
+    clean.click(chatgpt.Clean)
     gr.Examples(
         examples=["Write a poem about artificial intelligence",
                   "What could the future be like?",
@@ -55,4 +56,4 @@ with block:
         inputs=message
     )
 
-block.launch(debug=True) 
+block.launch(debug=True)