add o3-mini, chatgpt-4o-latest

app.py

@@ -103,6 +103,8 @@ languages = ["American English",
              "Portuguese"]
 
 models = ["gpt-4o",
+          "chatgpt-4o-latest",
+          "o3-mini",
           "claude-3-7-sonnet-latest",
           "claude-3-5-sonnet-20240620",
           "claude-3-5-sonnet-20241022",
@@ -386,7 +388,7 @@ Return the reformatted structure only. Do not add any preceding or trailing char
 
 <structure>/n{structure}</structure>"""
 
-gemini_prompt = """
+reasoning_prompt = """
 You are given information of a product, a reference structure, and a reference copy.
 Please analyze the structure, make a plan on how to follow the structure correctly.
 Use the tone of voice of the reference copy if available for the generated description.
@@ -616,8 +618,11 @@ Guidelines:
 
 
 def get_model(model_name, temperature=0.0):
-    if model_name.startswith("gpt"):
+    if model_name.startswith("gpt") \
+        or model_name.startswith("chatgpt"):
         chat = ChatOpenAI(model=model_name, max_tokens=4096, temperature=temperature)
+    elif model_name.startswith("o3"):
+        chat = openai.OpenAI()
     elif model_name.startswith("claude"):
         chat = ChatAnthropic(model_name=model_name, 
                              anthropic_api_key=os.environ["ANTHROPIC_API_KEY"], 
@@ -730,7 +735,7 @@ def generate(*data):
         if model.startswith("gemini"):
             if len((structure + copy).strip()) > 0:
                 messages = [
-                    gemini_prompt.format(min_length=min_length, 
+                    reasoning_prompt.format(min_length=min_length, 
                                          max_length=max_length, 
                                          key_features=key_features, 
                                          structure=structure, 
@@ -743,46 +748,70 @@ def generate(*data):
                 print("Gemini input: ", messages)
                 ri = chat.generate_content(messages)
                 print("Gemini response: ", ri)
-                response.append(ri)
+                response.append(ri.text)
         else:
-            if len((structure + copy).strip()) > 0:
-                if len(copy.strip()) > 0 and len(structure.strip()) > 0:    
-                    print('------------')
-                    print("Using both copy and structure")
-                    # print("Improving structure")
-                    # structure = improve_structure(chat=chat, structure=structure)
-                    messages = [
-                        SystemMessage(content=f"""You are a helpful assistant that writes about products for ecommerce websites. Make sure to write in {languages[i]} language."""),
-                        HumanMessage(content=struct_copy_prompt.format(nversions=nversions, min_length=min_length, max_length=max_length, key_features=key_features, structure=structure, copy=copy, included_phrases=included_phrases, excluded_phrases=excluded_phrases)),]
-                    
-                elif len(copy.strip()) > 0:
-                    print('------------')
-                    print("Using copy")
-                    messages = [
-                        SystemMessage(content=f"""You are a helpful assistant that writes about products for ecommerce websites. Make sure to write in {languages[i]} language."""),
-                        HumanMessage(content=copy_prompt.format(nversions=nversions, min_length=min_length, max_length=max_length, key_features=key_features, structure=structure, copy=copy, included_phrases=included_phrases, excluded_phrases=excluded_phrases)),]
+            if model.startswith("o3"):
+                messages = [
+                    {
+                        "role": "user",
+                        "content": reasoning_prompt.format(min_length=min_length, 
+                                         max_length=max_length, 
+                                         key_features=key_features, 
+                                         structure=structure, 
+                                         copy=copy, 
+                                         included_phrases=included_phrases, 
+                                         excluded_phrases=excluded_phrases)
+                    }
+                ]
 
-                elif len(structure.strip()) > 0:
+                ri = chat.chat.completions.create(
+                    model="o3-mini",
+                    messages=messages,
+                )
+
+                print("O3 input: ", messages)
+                print("O3 response: ", ri.choices[0].message.content)
+                response.append(ri.choices[0].message.content)
+            else:
+                if len((structure + copy).strip()) > 0:
+                    if len(copy.strip()) > 0 and len(structure.strip()) > 0:    
+                        print('------------')
+                        print("Using both copy and structure")
+                        # print("Improving structure")
+                        # structure = improve_structure(chat=chat, structure=structure)
+                        messages = [
+                            SystemMessage(content=f"""You are a helpful assistant that writes about products for ecommerce websites. Make sure to write in {languages[i]} language."""),
+                            HumanMessage(content=struct_copy_prompt.format(nversions=nversions, min_length=min_length, max_length=max_length, key_features=key_features, structure=structure, copy=copy, included_phrases=included_phrases, excluded_phrases=excluded_phrases)),]
+                        
+                    elif len(copy.strip()) > 0:
+                        print('------------')
+                        print("Using copy")
+                        messages = [
+                            SystemMessage(content=f"""You are a helpful assistant that writes about products for ecommerce websites. Make sure to write in {languages[i]} language."""),
+                            HumanMessage(content=copy_prompt.format(nversions=nversions, min_length=min_length, max_length=max_length, key_features=key_features, structure=structure, copy=copy, included_phrases=included_phrases, excluded_phrases=excluded_phrases)),]
+
+                    elif len(structure.strip()) > 0:
+                        print('------------')
+                        print("Using structure")
+                        # print("Improving structure")
+                        # structure = improve_structure(chat=chat, structure=structure)
+                        messages = [
+                            SystemMessage(content=f"""You are a helpful assistant that writes about products for ecommerce websites. Make sure to write in {languages[i]} language."""),
+                            HumanMessage(content=struct_prompt.format(nversions=nversions, min_length=min_length, max_length=max_length, key_features=key_features, structure=structure, copy=copy, included_phrases=included_phrases, excluded_phrases=excluded_phrases)),]
+                    print("Chat message", messages[1].content)
                     print('------------')
-                    print("Using structure")
-                    # print("Improving structure")
-                    # structure = improve_structure(chat=chat, structure=structure)
-                    messages = [
-                        SystemMessage(content=f"""You are a helpful assistant that writes about products for ecommerce websites. Make sure to write in {languages[i]} language."""),
-                        HumanMessage(content=struct_prompt.format(nversions=nversions, min_length=min_length, max_length=max_length, key_features=key_features, structure=structure, copy=copy, included_phrases=included_phrases, excluded_phrases=excluded_phrases)),]
-                print("Chat message", messages[1].content)
-                print('------------')
-                batch.append(messages)
+                    batch.append(messages)
 
-            response = chat.batch(batch)
+    if not(model.startswith("gemini") or model.startswith("o3")):
+        response = chat.batch(batch)
 
     descriptions = ""
 
     descriptions = []
     descriptions_post = []
         
-    if model.startswith("gemini"):
-        descriptions = [msg.text for msg in response]
+    if model.startswith("gemini") or model.startswith("o3"):
+        descriptions = response
         descriptions_post = [post_process(text=desc, 
                                           guidance_prompt=guidance_prompt, 
                                           language=languages[i],