Upload app.py

app.py

@@ -1,18 +1,21 @@
 # import the relevant packages
 import os
 import csv
+import requests
+import time
+import pandas as pd
 import openai
 from openai import OpenAI
 import gradio as gr
 import huggingface_hub
 from huggingface_hub import Repository
-from datetime import datetime
 
-client = OpenAI(
-    api_key=os.environ.get("API_TOKEN"),
-)
+api_key = os.environ.get("API_TOKEN")
+headers = {
+    'Authorization': 'Bearer ' + api_key,
+    'Content-Type': 'application/json'
+}
 
-# recording requests to Hugging Face dataset
 DATASET_REPO_URL = "https://huggingface.co/datasets/petcoblue/simulation_data"
 DATA_FILENAME = "user_agents.csv"
 DATA_FILE = os.path.join("data", DATA_FILENAME)
@@ -23,80 +26,80 @@ repo = Repository(
     local_dir="data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
 )
 
-# function to store the message to the Hugging Face site, if you would like to add more columns or information to track add it here
-def store_message(prompt: str, temperature: str, response: str):
-    if prompt:
-        with open(DATA_FILE, "a") as csvfile:
-            writer = csv.DictWriter(csvfile, fieldnames=["prompt", "temperature", "response", "time"])
-            writer.writerow(
-                {"prompt": prompt, 
-                 "temperature": temperature,
-                 "response": response,
-                 "time": str(datetime.now())}
-            )
-        commit_url = repo.push_to_hub()
-        print(commit_url)
+user_agents = pd.read_csv('./user_agents.csv')
+user_agents = user_agents.iloc[:,1:]
+user_batch = user_agents[:10]
 
-    return 
+def create_description(row):
+    description = (
+        f"Imagine that you are currently {int(row['age'])} years old. You have {int(row['num_pets'])} pets "
+        f"and spend an average of ${row['avg_spending']} on Petco purchases. "
+        f"Your engagement with Petco marketing has a score of {int(row['engagement_score'])}. "
+        f"You have an income level of {int(row['income_level'])} and "
+        f"regularly buy items from Petco every {int(row['purchase_regularity'])} months. "
+        f"It has been {int(row['time_since_last_purchase'])} days since your last purchase with Petco."
+    )
+    return description
 
-# defines who can enter the application with the secrets that are set up if used
-user_db = { 
-           os.environ["username"]: os.environ["password"],
-          }
+question = (
+    "Here are two images of Petco marketing emails:\n"
+    "- Image 0 is shown first.\n"
+    "- Image 1 is shown second.\n"
+    "Which email are you more likely to click through? Just answer with 0 for the first image or 1 for the second image.\n"
+    "Then, provide a list of up to five one-word characteristics of the email you chose that made you want to click through it. Separate each characteristic with a comma.\n\n"
+    "Example response:\n"
+    "1; Characteristics: Appealing, Sale, Bright, Simple, Exclusive\n"
+)
 
-# the OpenAI function to use the given prompt to the model defined.
-def mr_bot(prompt, tokens, temperature):
-    response = client.chat.completions.create(
-        model="gpt-3.5-turbo",
-        messages=[{"role": "user", "content": prompt}],
-        max_tokens=tokens,
-        temperature=temperature,
-        )
-    store_message(prompt=prompt, temperature=temperature, response=response.choices[0].message.content)
-    
-    return response.choices[0].message.content
+def query_agent(description, question, image0, image1):
+    image0_path = os.path.join(image_directory, image0)
+    image1_path = os.path.join(image_directory, image1)
+    base64_image0 = encode_image(image0_path)
+    base64_image1 = encode_image(image1_path)
 
+    payload = {
+        "model": "gpt-4-vision-preview",
+        "messages": [
+            {"role": "system", "content": description},
+            {
+                "role": "user",
+                "content": [
+                    {"type": "text", "text": question},
+                    {"type": "image", "image_url": f"data:image/jpeg;base64,{base64_image0}"},
+                    {"type": "image", "image_url": f"data:image/jpeg;base64,{base64_image1}"}
+                ]
+            }
+        ],
+        "max_tokens": 300,
+        "logprobs": True,
+        "top_logprobs": 1
+    }
 
-# Gradio App interface and login set up
-description = "Use the examples below to see how different prompts generate different results. Use the examples to see how different prompting " \
-              "techniques can enhance your prompts. They are split into three sections, (creative writing, problem solving in business, and " \
-              "philosophical debate) with three different levels of prompt engineering, (basic, intermediate, and advanced)."
+    for attempt in range(3):
+        try:
+            response = requests.post("https://api.openai.com/v1/chat/completions", headers=headers, json=payload)
+            if response.status_code == 200:
+                data = response.json()
+                preference = data['choices'][0]['message']['content']
+                top_logprobs = data['choices'][0]['logprobs']['content'][0]['top_logprobs']
+                return preference, top_logprobs
+            else:
+                print(f"HTTP Error {response.status_code} on attempt {attempt + 1}")
+        except requests.exceptions.RequestException as e:
+            print(f"Request failed on attempt {attempt + 1}: {e}")
+            time.sleep(1)
+    else:
+        print(f"Failed to analyze {image0} and {image1} after 3 attempts.")
+        return None, None
 
-demo = gr.Interface(fn=mr_bot, 
-                    inputs=[gr.Textbox(placeholder="Select an example", interactive=False), gr.Slider(1, 800, default=800), gr.Slider(0.1, 1, default=0.75)], 
-                    outputs="text",
-                    title="Intro to AI, Prompt Engineering",
-                    description=description,
-                    cache_examples=False,
-                    examples = [
-                                            # Creative Writing
-                        # Basic Prompt:
-                        ["Write a story about a city.", 800, .90],
-                        # Intermediate Prompt:
-                        ["Write a story set in a futuristic city where technology controls everything.", 800, .90],
-                        # Advanced Prompt:
-                        ["Write a gripping tale set in a futuristic city named Neo-Philly, where an underground movement " \
-                         "is fighting against the oppressive control of AI over human life, focusing on a protagonist who was once a loyal AI prompt engineer.", 800, .90],
+description = "Upload two images of emails and see which is generally preferred by Petco customers!"
 
-                                            # Problem Solving in Business
-                        # Basic Prompt:
-                        ["A company needs to improve its profits.", 800, .90],
-                        # Intermediate Prompt:
-                        ["A retail company is seeing a decline in profits due to online competition. What can they do?", 800, .90],
-                        # Advanced Prompt:
-                        ["Devise a comprehensive strategy for a brick-and-mortar retail company that has been losing market share to e-commerce giants. " \
-                         "Focus on innovative in-store experiences, digital integration, and customer loyalty programs to regain competitiveness.", 800, .90],
-
-                                            # Philosophical Debate
-                        # Basic Prompt:
-                        ["Discuss the concept of happiness.", 800, .90],
-                        # Intermediate Prompt:
-                        ["Debate whether true happiness can be achieved through material wealth or if it's found in intangible experiences.", 800, .90],
-                        # Advanced Prompt:
-                        ["Critically analyze the philosophical arguments for and against the notion that true happiness is derived more from personal " \
-                         "fulfillment and self-actualization than from material possessions, considering perspectives from both Eastern and " \
-                         "Western philosophies.", 800, .90],
-                    ]
+demo = gr.Interface(fn=query_agent, 
+                    inputs=[gr.UploadButton("Click to Upload Email 0", file_types=["image"], file_count="1"), 
+                            gr.UploadButton("Click to Upload Email 1", file_types=["image"], file_count="1")], 
+                    outputs="text",
+                    title="Pairwise Simulation of Petco Email Preference",
+                    description=description
                     )
 
 if __name__ == "__main__":