ported to chatgpt api

app.py

@@ -60,10 +60,9 @@ DEFAULT_PROMPT = ("You are a Leg Lift Classifier. There is an image of a through
 
 def predict(img, prompt):
     # Encode the Pillow image in base64
-    buffer = BytesIO()
-    img.save(buffer, format="PNG")  # Save the image in PNG format (can change to JPEG if needed)
-    buffer.seek(0)
-    encoded_image = base64.b64encode(buffer.read()).decode('utf-8')
+    image_bytes = BytesIO()
+    img.save(image_bytes, format='PNG')  # Save as PNG (can be changed to JPEG)
+    image_bytes.seek(0)  # Rewind the BytesIO buffer
 
     # Combine prompt and image
     # system_prompt = (
@@ -73,14 +72,15 @@ def predict(img, prompt):
 
     messages = [
         {"role": "system", "content": prompt},
-        {"role": "user", "content": f"Image (base64): {encoded_image}"}
+        {"role": "user", "content": f"Image (base64): {image_bytes}"}
     ]
 
     # Make API call
     try:
         response = openai.ChatCompletion.create(
-            model="gpt-4",
-            messages=messages
+            model="gpt-4-vision",
+            messages=messages,
+             files=[("file", image_bytes)]
         )
         return response['choices'][0]['message']['content']
     except Exception as e: