Update app.py

app.py

@@ -32,8 +32,6 @@ 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))
 
-from openai import OpenAI
-
 # Initialize the OpenAI client
 client = OpenAI()
 
@@ -45,7 +43,7 @@ def process_chat(user_text):
     try:
         # Use the OpenAI client for creating a chat completion
         response = client.chat.completions.create(
-            model="gpt-4",
+            model="gpt-4o-mini",
             messages=[
                 {"role": "system", "content": "You are a helpful assistant named Diane specializing in digital art advice."},
                 {"role": "user", "content": user_text},
@@ -57,8 +55,8 @@ def process_chat(user_text):
         for chunk in response:
             # 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
+            token = getattr(delta, "content", None)  # Safely get the "content" field
+            if token:  # Only process non-None tokens
                 response_text += token
                 yield response_text
 
@@ -66,7 +64,6 @@ def process_chat(user_text):
         yield f"❌ An error occurred: {str(e)}"
 
 
-
 # Function to analyze image contrast
 def analyze_contrast_opencv(image_path):
     img = cv2.imread(image_path, cv2.IMREAD_GRAYSCALE)