Update app.py

app.py

@@ -1,23 +1,25 @@
 import openai
 import gradio as gr
 
+model_id = "gpt-3.5-turbo"
+
 # Getting responses using the OpenAI API
-def answer_chatgpt(api_key, message, history):
+def answer_chatgpt(api_key, message, history, conversation=[]):
   history = history or []
   lista = list(sum(history, ()))
   lista.append(message)
+  prompt = f"{message}"
+  conversation.append({"role": "user", "content": f"{prompt}"})
   # OPENAI API KEY
   openai.api_key = api_key
-  prompt = f"{message}"
   response = openai.ChatCompletion.create(
-  model="gpt-3.5-turbo",
-  messages=[
-        {"role": "user", "content": f"{prompt}"},
-    ]
+  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
       
 # User input