Update app.py

app.py

@@ -1,59 +1,78 @@
+
+
+import os
+import time
 import openai
 import gradio as gr
 from gtts import gTTS
 from io import BytesIO
 from IPython.display import Audio, display
-import os
-import time
 
 # Set OpenAI API key
 openai.api_key = os.environ.get("OPENAI_API_KEY")
 
 # Set OpenAI GPT-3 model
-MODEL = "gpt-3.5-turbo"
+MODEL = "text-davinci-002"
 
 
+# Initialize chat history as an empty list
+chat_history = []
 
-def chat(input_text, history):
-    global chat_history
-    if len(input_text) == 0:
-        return "", None
+# Define function to generate speech from text using Google Text-to-Speech (gTTS)
+def text_to_speech(text):
+    tts = gTTS(text=text)
+    mp3 = BytesIO()
+    tts.write_to_fp(mp3)
+    mp3.seek(0)
+    display(Audio(mp3, autoplay=True))
+
+# Define function to get chatbot response
+def chat(text):
+    # Append user input to chat history
+    chat_history.append(f"User: {text}")
+    
+    # Use OpenAI's GPT-3.5 model to generate chatbot response
     response = openai.Completion.create(
-      model=MODEL,
-      prompt=history + input_text,
-      max_tokens=1024,
-      n=1,
-      stop=None,
-      temperature=0.5,
-    )
-    message = response.choices[0].text.strip()
-    chat_history += f"\nUser: {input_text}\nAI: {message}\n"
-    # create speech output using gTTS
-    speech_output = gTTS(message)
-    # create byte buffer to store speech output
-    speech_buffer = BytesIO()
-    speech_output.write_to_fp(speech_buffer)
-    speech_buffer.seek(0)
-    # display speech output using IPython.display
-    display(Audio(speech_buffer.read(), format='audio/mp3'))
-    return message, speech_buffer.getvalue()
+        model=MODEL,
+        prompt=f"Conversation with user:\n{'\n'.join(chat_history)}\nChatbot:",
+        temperature=0.5,
+        max_tokens=1024,
+        n=1,
+        stop=None,
+        frequency_penalty=0,
+        presence_penalty=0
+    ).choices[0].text.strip()
+    
+    # Append chatbot response to chat history
+    chat_history.append(f"Chatbot: {response}")
+    
+    # Generate speech from chatbot response
+    text_to_speech(response)
     
-def run():
+    return response
+
+# Define function to clear chat history
+def clear_chat():
     global chat_history
-    chat_history = ""
-    interface = gr.Interface(
-        fn=chat,
-        inputs=["text", gr.inputs.Dropdown(["en-US-Wavenet-A", "en-US-Wavenet-B", "en-US-Wavenet-C", "en-US-Wavenet-D"])],
-        outputs=["text", "audio"],
-        layout="vertical",
-        title="OpenAI Chatbot with Speech Output",
-        description="Chat with the OpenAI GPT-3 chatbot and get speech output of the bot's response."
-    )
-
-    interface.launch()
-    while True:
-        time.sleep(0.1)
-        with open("chat_history.txt", "w") as f:
-            f.write(chat_history)
+    chat_history = []
+
+# Define interface
+interface = gr.Interface(
+    chat,
+    inputs=["text", gr.inputs.Voice()],
+    outputs=["text", gr.outputs.Voice()],
+    title="Chatbot with OpenAI's GPT-3.5 Model",
+    description="An interactive chatbot using OpenAI's GPT-3.5 model with chat persistence and voice inputs/outputs.",
+    theme="default",
+    layout="vertical",
+    allow_flagging=False,
+    allow_screenshot=False,
+    allow_download=False,
+    show_input=True,
+    show_output=True
+)
+
+# Run interface
+interface.launch()