Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import gradio as gr
 import openai
+from openai import OpenAI
 from PIL import Image, ImageEnhance
 import cv2
 import torch
@@ -31,27 +32,32 @@ object_labels = [
 EXAMPLE_IMAGE_URL = "https://www.watercoloraffair.com/wp-content/uploads/2023/04/monet-houses-of-parliament-low-key.jpg"  # Square example image
 example_image = Image.open(BytesIO(requests.get(EXAMPLE_IMAGE_URL).content))
 
+# Initialize OpenAI client
+client = OpenAI()
+
 def process_chat(user_text):
     if not user_text.strip():
         yield "⚠️ Please enter a valid question."
         return
 
     try:
-        # Use ChatCompletion API for chat models
-        response = openai.ChatCompletion.create(
-            model="gpt-4",
+        # Use the client to create a completion
+        response = client.chat.completions.create(
+            model="gpt-4o",
             messages=[
                 {"role": "system", "content": "You are a helpful assistant named Diane specializing in digital art advice."},
                 {"role": "user", "content": user_text},
             ],
             stream=True  # Enable streaming
         )
+
         response_text = ""
         for chunk in response:
-            if "choices" in chunk and chunk.choices[0].delta.get("content"):
+            if chunk.choices[0].delta.get("content"):
                 token = chunk.choices[0].delta["content"]
                 response_text += token
                 yield response_text
+
     except Exception as e:
         yield f"❌ An error occurred: {str(e)}"