OpenAI basic error handling

app.py

@@ -202,32 +202,35 @@ def get_openAI_tags(image_urls):
     imageList = []
     for image in image_urls:
         imageList.append({"type": "image_url", "image_url": {"url": image}})
-    
-    openai_response = client.chat.completions.create(
-        model="gpt-4o",
-        messages=[
-            {
-            "role": "system",
-            "content": [
+    try:
+        openai_response = client.chat.completions.create(
+            model="gpt-4o",
+            messages=[
+                {
+                "role": "system",
+                "content": [
+                    {
+                    "type": "text",
+                    "text": "You're a tagging assistant, you will help label and tag product pictures for my online e-commerce platform. Your tasks will be to return which angle the product images were taken from. You will have to choose from 'full-body', 'half-body', 'side', 'back', or 'zoomed' angles. You should label each of the images with one of these labels depending on which you think fits best (ideally, every label should be used at least once, but only if there are 5 or more images), and should respond with an unformatted dictionary where the key is a string representation of the url index of the url and the value is the assigned label."
+                    }
+                ]
+                },
                 {
-                "type": "text",
-                "text": "You're a tagging assistant, you will help label and tag product pictures for my online e-commerce platform. Your tasks will be to return which angle the product images were taken from. You will have to choose from 'full-body', 'half-body', 'side', 'back', or 'zoomed' angles. You should label each of the images with one of these labels depending on which you think fits best (ideally, every label should be used at least once, but only if there are 5 or more images), and should respond with an unformatted dictionary where the key is a string representation of the url index of the url and the value is the assigned label."
-                }
-            ]
-            },
-            {
-            "role": "user",
-            "content": imageList
-            },
-        ],
-        temperature=1,
-        max_tokens=500,
-        top_p=1,
-        frequency_penalty=0,
-        presence_penalty=0
-    )
-    response = json.loads(openai_response.choices[0].message.content)
-    return response
+                "role": "user",
+                "content": imageList
+                },
+            ],
+            temperature=1,
+            max_tokens=500,
+            top_p=1,
+            frequency_penalty=0,
+            presence_penalty=0
+        )
+        response = json.loads(openai_response.choices[0].message.content)
+        return response
+    except Exception as e:
+        print(f"OpenAI API Error: {e}")
+        return {}
 
 
 @spaces.GPU