add gpt-4o

app.py

@@ -55,6 +55,7 @@ languages = ["American English",
              "Portuguese"]
 
 models = ["gpt-4-turbo", 
+          "gpt-4o",
           "gpt-3.5-turbo", 
           "claude-3-sonnet-20240229", 
           "claude-3-opus-20240229", 
@@ -105,7 +106,7 @@ def detect_features(image_paths, garment_type, language="English"):
         }
 
         payload = {
-        "model": "gpt-4-vision-preview",
+        "model": "gpt-4o",
         "messages": [
             {
             "role": "user",
@@ -138,6 +139,8 @@ def detect_features(image_paths, garment_type, language="English"):
 
         response = requests.post("https://api.openai.com/v1/chat/completions", headers=headers, json=payload)
 
+        print(response)
+
         response = response.json()
         print("image features", response["choices"][0]['message']['content'])
         jresponse = get_json(response["choices"][0]['message']['content'])
@@ -150,6 +153,7 @@ def detect_features(image_paths, garment_type, language="English"):
 
 def generate(*data):
     global visible
+    print("visible", visible)
 
     feature, image, garment_type, model, temperature = data[:5]
     struct_ref = data[5:]
@@ -179,7 +183,7 @@ def generate(*data):
         messages = [
             SystemMessage(content=f"""You are a helpful assistant that writes about products for ecommerce websites."""),
             HumanMessage(content=f"""Write a product description with the following features.
-                                Make sure that the description follows the structure of the reference structure. 
+                                Make sure that the description follows the structure of the reference structure. The first line in the reference structure include the document type and the language code. The next lines define the structure of the document to be generated.
                                 Make sure to use markdown format for the output.
                                 Make sure that the entire output is written entirely in language defined in the reference structure. 
                                 Use language that is suitable for the type of document specified in the reference structure. 
@@ -256,6 +260,5 @@ with gr.Blocks() as demo:
         advanced.click(show_advanced, inputs=[], outputs=[model, temperature])
         
 
-
 if __name__ == '__main__':
     demo.launch()