update to .env and using gpt-3.5-turbo

app.py

@@ -1,45 +1,35 @@
 import gradio as gr
-from transformers import pipeline, set_seed
+import openai
+import os
 
-# Initialize the chat model pipeline
-chat = pipeline('text-generation', model='gpt-3.5-turbo', use_auth_token='Your_Hugging_Face_API_Token_Here')
+# Fetch your OpenAI API key from an environment variable
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
-def chat_with_chatgpt(user_message, system_message, chat_history):
-    set_seed(42)  # Optional: for consistent results
-    
-    # Combine system message, chat history, and current user message for context
-    if system_message not in chat_history:  # Include system message only at the beginning
-        input_text = f"{system_message}\n{chat_history} You: {user_message}"
-    else:
-        input_text = f"{chat_history} You: {user_message}"
-    
-    # Generate response from ChatGPT
-    response = chat(input_text, max_length=1000)
-    generated_text = response[0]['generated_text']
-    
-    # Extract only ChatGPT's latest response
-    new_response = generated_text[len(input_text):].strip()
-    
-    # Update chat history
-    new_chat_history = f"{chat_history} You: {user_message}\nChatGPT: {new_response}\n"
-    
-    return new_chat_history, new_chat_history  # Return updated chat history for both display and state
+if openai.api_key is None:
+    raise ValueError("OPENAI_API_KEY environment variable not set.")
 
-# Create the Gradio interface
+def chat_with_gpt(system_message, user_message):
+    conversation = f"System: {system_message}\nUser: {user_message}"
+    response = openai.Completion.create(
+        model="gpt-3.5-turbo",
+        prompt=conversation,
+        temperature=0.5,
+        max_tokens=500,
+        frequency_penalty=0.0,
+        presence_penalty=0.0,
+    )
+    return response.choices[0].text.strip()
+
+# Define the interface
 iface = gr.Interface(
-    fn=chat_with_chatgpt,
+    fn=chat_with_gpt,
     inputs=[
-        gr.inputs.Textbox(label="Your Message"),
-        gr.inputs.Textbox(label="System Message (Enter only before starting the chat)", lines=2),
-        gr.State(label="Chat History")
-    ],
-    outputs=[
-        gr.outputs.Textbox(label="Chat History"),
-        gr.outputs.Textbox(label="New Chat History", visible=False)
+        gr.inputs.Textbox(lines=2, placeholder="System Message Here..."),
+        gr.inputs.Textbox(lines=5, placeholder="Your Message Here..."),
     ],
-    title="Chat with ChatGPT 3.5",
-    description="Start with a system message and then continue chatting like in ChatGPT.",
-)
-
-if __name__ == "__main__":
-    iface.launch()
+    outputs="text",
+    title="Chat with GPT-3.5",
+    description="This Gradio app lets you chat with OpenAI's GPT-3.5 model. Enter a system message for initial context, and then chat as you would with a human.",
+    theme="default",  # or "huggingface" for Hugging Face theme
+    allow_flagging="never",
+).launch()

requirements.txt

@@ -1,2 +1,3 @@
-gradio
-transformers
+os
+openai
+gradio