Update app.py

app.py

@@ -89,27 +89,33 @@ def enhance_contrast(image):
     enhanced_image.save(enhanced_path)
     return enhanced_path
 
-# Function to provide additional suggestions with streaming
 def provide_suggestions_streaming(object_identified):
     if not object_identified:
-        yield "Sorry, I couldn't find an object in your artwork. Try a different image."
+        yield "⚠️ Sorry, I couldn't identify an object in your artwork. Try uploading a different image."
         return
 
-    stream = openai.ChatCompletion.create(
-        model="gpt-4",
-        messages=[
-            {"role": "system", "content": "You are an expert digital art advisor."},
-            {"role": "user", "content": f"Suggest ways to improve a digital artwork featuring a {object_identified}."}
-        ],
-        stream=True
-    )
+    try:
+        # Use the OpenAI client for suggestions
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {"role": "system", "content": "You are an expert digital art advisor."},
+                {"role": "user", "content": f"Suggest ways to improve a digital artwork featuring a {object_identified}."},
+            ],
+            stream=True  # Enable streaming
+        )
 
-    response = ""
-    for chunk in stream:
-        if chunk.choices[0].delta.get("content"):
-            token = chunk.choices[0].delta["content"]
-            response += token
-            yield response
+        response_text = ""
+        for chunk in response:
+            # Extract the content safely
+            delta = chunk.choices[0].delta  # Get the delta object
+            token = getattr(delta, "content", None)  # Safely access the "content" field
+            if token:  # Only process non-None tokens
+                response_text += token
+                yield response_text
+
+    except Exception as e:
+        yield f"❌ An error occurred while providing suggestions: {str(e)}"
 
 # Main image processing function
 def process_image(image):