added state

app.py

@@ -28,17 +28,24 @@ def transcribe_speech(filepath):
 load_dotenv()
 openai.api_key = os.environ['OPENAI_API_KEY']
 
-def query_chatgpt(text):
-    messages = []
-    messages.append({'role': 'user', 'content': '{}'.format(text)})
-    print("Preguntando "+text)
+
+def clear_chat():
+     global chat_history
+     chat_history=[]
+
+
+def query_chatgpt(message,chat_history):
+    chat_history.append({'role': 'user', 'content': '{}'.format(message)})
+    print("Preguntando "+message)
+    print("historial", chat_history)
     response = openai.ChatCompletion.create(
             model="gpt-3.5-turbo",
-            messages= messages,
+            messages= chat_history,
             temperature=0.5,
             max_tokens=256
             ).choices[0].message.content
-    return response
+    chat_history.append({'role': 'assistant', 'content': '{}'.format(response)})
+    return response, chat_history
 
 
 
@@ -65,22 +72,29 @@ from tts import synthesize
 #     wav,rate = TTSHubInterface.get_prediction(task, model, generator, sample)
 #     return wav,rate
 
-def answer_question(filepath):
+def answer_question(filepath,chat_history):
     transcription = transcribe_speech(filepath)
-    response = query_chatgpt(transcription)
+    response,chat_history = query_chatgpt(transcription,chat_history)
+    print("historial",chat_history)
     # audio = synthesise(response)
-    print(response)
     # audio, rate = syn_facebookmms(response)
     rate,audio = synthesize(response,1,"spa")
     print(audio)
     return rate,audio
 
+def reset_state(chat_history):
+    chat_history = []
+    return chat_history
+
 
 import gradio as gr
 with gr.Blocks() as demo:
+    chat_history = gr.State([])
     entrada = gr.Audio(source="microphone",type="filepath")
     boton = gr.Button("Responder")
-
+    button = gr.Button("Reset State")
     salida = gr.Audio()
-    boton.click(answer_question,entrada,salida)
+    boton.click(answer_question,[entrada,chat_history],salida)
+    button.click(reset_state,chat_history,chat_history)
+  
 demo.launch(debug=True)