Update app.py

app.py

@@ -113,7 +113,35 @@ def get_predicted_attributes(image_urls, category):
     return result
 
 
-
+def get_openAI_tags(image_urls):
+    # Create list containing JSONs of each image URL
+    imageList = []
+    for image in image_urls:
+        imageList.append({"type": "image_url", "image_url": {"url": image}})
+    
+    return 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 nothing but the labels separated by a comma in the order of the images without any other text. You should label every picture, no more, no less."
+                }
+            ]
+            },
+            {
+            "role": "user",
+            "content": imageList
+            },
+        ],
+        temperature=1,
+        max_tokens=500,
+        top_p=1,
+        frequency_penalty=0,
+        presence_penalty=0
+    )
 
 # Define the Gradio interface with the updated components
 iface = gr.Interface(