Update app.py

app.py

@@ -1,39 +1,44 @@
 
+import openai
+import os
+from dotenv import load_dotenv, find_dotenv
+
+_ = load_dotenv(find_dotenv()) # read local .env file
+
+openai.api_key  = os.getenv('OPENAI_API_KEY')
+
+
+def get_completion_from_messages(messages, 
+                                 model="gpt-3.5-turbo", 
+                                 temperature=0, 
+                                 max_tokens=500):
+    response = openai.chat.completions.create( # Use openai.chat.completions.create instead of openai.ChatCompletion.create
+        model=model,
+        messages=messages,
+        temperature=temperature, 
+        max_tokens=max_tokens,
+    )
+    return response.choices[0].message.content
 import gradio as gr
-import google.generativeai as genai
-
-# Replace with your actual API key
-genai.configure(api_key='GOOGLE_API_KEY')
-
-chat_history = []  # Initialize chat history
-
-def generate_text(prompt):
-  """Generates text using Gemini Pro and updates chat history."""
-
-  global chat_history
-
-  chat_history.append({"role": "user", "content": prompt})  # Add user message
-
-  response = genai.text.generate_text(
-      model='models/gemini-pro',
-      prompt="\n".join([f"{m['role']}: {m['content']}" for m in chat_history]),
-      temperature=0.7,
-      max_output_tokens=1024,
-  )
-
-  chat_history.append({"role": "assistant", "content": response.text})  # Add assistant message
-
-  # Format chat history for display
-  chat_display = "\n".join([f"{m['role']}: {m['content']}" for m in chat_history])
-
-  return chat_display
-
-iface = gr.Interface(
-    fn=generate_text,
-    inputs=gr.Textbox(lines=2, placeholder="Enter your prompt here..."),
-    outputs="text",
-    title="Gemini Pro Chatbot with History",
-    description="Ask me anything!",
-)
-
-iface.launch()
+def generate(input, slider):
+    messages =  [
+        {'role':'user',
+         'content': f"{input}"},
+    ]
+    
+    output = get_completion_from_messages(messages)
+    return output
+
+demo = gr.ChatInterface(
+    fn=generate,
+    chatbot=gr.Chatbot(height=300),
+    textbox=gr.Textbox(placeholder="Ask me a yes or no question", container=False, scale=7),
+    title="Your Helper CHAT Bot",
+    description="Ask 'Yes Man' any question",
+    theme="soft",
+    examples=["Hello", "Am I cool?", "Are tomatoes vegetables?"],
+    cache_examples=True,
+    #retry_btn=None,
+   #undo_btn="Delete Previous",
+    #clear_btn="Clear",
+).launch()