Update app.py

app.py

@@ -1,26 +1,44 @@
-import gradio as gr
 import os
 import openai
+import gradio as gr
 
 # Replace 'your_api_key_here' with your actual OpenAI API key
 api_token = os.getenv("oaikey")
-openai.api_key = api_token
 
-def ask_openai(prompt):
+# Set your OpenAI API key
+openai.api_key = os.getenv(api_token)
+
+# Initialize chat history
+messages = [{"role": "system", "content": "You are a helpful assistant."}]
+
+def chatbot(user_input):
+    # Append user input to the message history
+    messages.append({"role": "user", "content": user_input})
+    
+    # Generate response using OpenAI's GPT-3.5-turbo
     response = openai.ChatCompletion.create(
-        model="gpt-4",  # or any other suitable model
-        messages=[{"role": "user", "content": prompt}]
+        model="gpt-3.5-turbo",
+        messages=messages,
+        temperature=0.7,
+        max_tokens=150,
+        top_p=1,
+        frequency_penalty=0,
+        presence_penalty=0
     )
-    return response['choices'][0]['message']['content']
+    
+    # Extract the assistant's reply
+    assistant_reply = response.choices[0].message["content"].strip()
+    
+    # Append assistant's reply to the message history
+    messages.append({"role": "assistant", "content": assistant_reply})
+    
+    return assistant_reply
 
-# Creating the Gradio Interface
-iface = gr.Interface(
-    fn=ask_openai,
-    inputs="text",
-    outputs="text",
-    title="OpenAI Chatbot",
-    description="A simple chatbot using OpenAI's GPT model."
-)
+# Create Gradio interface
+with gr.Blocks() as demo:
+    chatbot_ui = gr.ChatInterface(fn=chatbot, title="GPT-3.5 Turbo Chatbot")
 
+# Launch the Gradio app
 if __name__ == "__main__":
-    iface.launch()
+    demo.launch()
+