Update

app.py

@@ -1,10 +1,9 @@
 import streamlit as st
-import requests
+from openai import OpenAI
 from dotenv import load_dotenv
 import os
 import uuid
 from datetime import datetime
-import openai
 import time
 import random
 
@@ -35,8 +34,13 @@ if "selected_model" not in st.session_state:
 
 # Get OpenAI API key from environment or user input
 openai_api_key = os.getenv("OPENAI_API_KEY_NEW_3")
+if not openai_api_key:
+    openai_api_key = st.sidebar.text_input("Enter OpenAI API Key", type="password")
+
+# Initialize OpenAI client
+client = None
 if openai_api_key:
-    openai.api_key = openai_api_key
+    client = OpenAI(api_key=openai_api_key)
 
 # Available models with descriptions and token limits
 AVAILABLE_MODELS = {
@@ -110,7 +114,7 @@ AVAILABLE_MODELS = {
 
 # Function to call OpenAI API
 def get_ai_response(prompt, history):
-    if not openai_api_key:
+    if not client:
         return get_demo_response(prompt)
     
     try:
@@ -124,27 +128,17 @@ def get_ai_response(prompt, history):
         model = st.session_state.selected_model
         model_config = AVAILABLE_MODELS.get(model, AVAILABLE_MODELS["gpt-3.5-turbo"])
         
-        response = openai.ChatCompletion.create(
+        response = client.chat.completions.create(
             model=model,
             messages=messages,
             temperature=model_config["temperature"],
             max_tokens=model_config["output_tokens"],
             stream=False
         )
-        return response['choices'][0]['message']['content']
+        return response.choices[0].message.content
     
-    except openai.error.AuthenticationError as auth_err:
-        return f"Authentication error: {str(auth_err)}. Check your API key."
-    except openai.error.InvalidRequestError as inv_err:
-        return f"Invalid request: {str(inv_err)}. The model '{model}' might not be available with your API key."
-    except openai.error.APIConnectionError as conn_err:
-        return f"API connection error: {str(conn_err)}. Please check your network and try again."
-    except openai.error.RateLimitError as rate_err:
-        return f"Rate limit exceeded: {str(rate_err)}. Please wait before trying again."
-    except openai.error.OpenAIError as openai_err:
-        return f"OpenAI error: {str(openai_err)}. An unexpected error occurred."
     except Exception as e:
-        return f"An unexpected error occurred: {str(e)}."
+        return f"An error occurred: {str(e)}."
 
 # Function to create a new conversation
 def create_new_chat():