Update app.py

app.py

@@ -31,28 +31,29 @@ 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))
 
-# Updated process_chat function
 def process_chat(user_text):
     if not user_text.strip():
-        yield "⚠️ Please enter a question."
+        yield "⚠️ Please enter a valid question."
         return
 
-    # Use openai.Chat for the new API
-    response = openai.Chat.create(
-        model="gpt-4",
-        messages=[
-            {"role": "system", "content": "You are a helpful assistant named Diane specializing in digital art advice."},
-            {"role": "user", "content": user_text}
-        ]
-    )
-
-    # Stream the response content
-    response_text = ""
-    for chunk in response:
-        if "choices" in chunk and chunk["choices"][0].get("delta", {}).get("content"):
-            token = chunk["choices"][0]["delta"]["content"]
-            response_text += token
-            yield response_text
+    try:
+        # Streaming responses from OpenAI API
+        response = openai.Chat.create(
+            model="gpt-4",
+            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 chunk.get("choices") and chunk["choices"][0].get("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)}"
 
 
 # Function to analyze image contrast