Upload app.py

app.py

@@ -3,6 +3,7 @@ import os
 import csv
 import requests
 import time
+import base64
 import pandas as pd
 import openai
 from openai import OpenAI
@@ -21,6 +22,10 @@ user_agents = dataset['train'].to_pandas()
 user_agents = user_agents.iloc[:,1:]
 user_batch = user_agents[:10]
 
+def encode_image(image_path):
+    with open(image_path, "rb") as image_file:
+        return base64.b64encode(image_file.read()).decode('utf-8')
+
 def create_description(row):
     description = (
         f"Imagine that you are currently {int(row['age'])} years old. You have {int(row['num_pets'])} pets "
@@ -43,10 +48,8 @@ question = (
 )
 
 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)
+    base64_image0 = encode_image(image0)
+    base64_image1 = encode_image(image1)
 
     payload = {
         "model": "gpt-4-vision-preview",