Update app.py

app.py

@@ -1,3 +1,4 @@
+import ast
 import json
 import spaces
 import requests
@@ -6,14 +7,19 @@ import gradio as gr
 from PIL import Image
 from io import BytesIO
 from turtle import title
-from transformers import pipeline
-import ast
+from openai import OpenAI
 from collections import Counter
+from transformers import pipeline
+
+client = OpenAI()
 
 pipe = pipeline("zero-shot-image-classification", model="patrickjohncyh/fashion-clip")
 
 color_file_path = 'color_config.json'
 attributes_file_path = 'attributes_config.json'
+import os
+OPENAIKEY = os.getenv("OPENAI_KEY")
+
 
 # Open and read the COLOR JSON file
 with open(color_file_path, 'r') as file:
@@ -36,7 +42,8 @@ def shot(input, category):
             "sub":subColour,
             "score":round(score*100,2)
         }, 
-        "attributes":common_result
+        "attributes":common_result,
+        "image_mapping":openai_parsed_response
     }
 
 
@@ -119,7 +126,7 @@ def get_openAI_tags(image_urls):
     for image in image_urls:
         imageList.append({"type": "image_url", "image_url": {"url": image}})
     
-    return client.chat.completions.create(
+    openai_response = client.chat.completions.create(
         model="gpt-4o",
         messages=[
             {
@@ -142,6 +149,8 @@ def get_openAI_tags(image_urls):
         frequency_penalty=0,
         presence_penalty=0
     )
+    response= json.loads(openai_response.choices[0].message.content)
+    return response
 
 # Define the Gradio interface with the updated components
 iface = gr.Interface(