Update app.py

app.py

@@ -32,7 +32,9 @@ 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
+from openai import OpenAI
+
+# Initialize the OpenAI client
 client = OpenAI()
 
 def process_chat(user_text):
@@ -41,9 +43,9 @@ def process_chat(user_text):
         return
 
     try:
-        # Use the client to create a completion
+        # Use the OpenAI client for creating a chat completion
         response = client.chat.completions.create(
-            model="gpt-4o",
+            model="gpt-4",
             messages=[
                 {"role": "system", "content": "You are a helpful assistant named Diane specializing in digital art advice."},
                 {"role": "user", "content": user_text},
@@ -53,8 +55,10 @@ def process_chat(user_text):
 
         response_text = ""
         for chunk in response:
-            if chunk.choices[0].delta.get("content"):
-                token = chunk.choices[0].delta["content"]
+            # Extract the content correctly
+            delta = chunk.choices[0].delta  # Get the delta object
+            if hasattr(delta, "content"):  # Check if the delta has "content"
+                token = delta.content
                 response_text += token
                 yield response_text