Update app.py via AI Editor

app.py

@@ -1,4 +1,4 @@
-# Generated code appears hereimport os
+import os
 import base64
 import io
 import dash
@@ -10,7 +10,6 @@ import logging
 from docx import Document
 import requests
 import mimetypes
-import os
 
 logging.basicConfig(
     level=logging.INFO,
@@ -67,7 +66,7 @@ def guess_mime_type(filename):
 
 def upload_to_openai_file(decoded_bytes, filename):
     try:
-        url = "https://api.openai.com/v1/uploads"
+        url = "https://api.openai.com/v1/files"
         mime_type = guess_mime_type(filename)
         headers = {
             "Authorization": f"Bearer {OPENAI_API_KEY}"
@@ -76,27 +75,24 @@ def upload_to_openai_file(decoded_bytes, filename):
             "file": (filename, io.BytesIO(decoded_bytes), mime_type)
         }
         data = {
-            "bytes": str(len(decoded_bytes)),
-            "filename": filename,
-            "mime_type": mime_type,
             "purpose": "assistants"
         }
-        logging.info(f"Uploading file to OpenAI /v1/uploads: {filename}, size={len(decoded_bytes)}, mime_type={mime_type}")
+        logging.info(f"Uploading file to OpenAI /v1/files: {filename}, size={len(decoded_bytes)}, mime_type={mime_type}")
         resp = requests.post(url, headers=headers, files=files, data=data)
         if resp.status_code == 200:
             resp_json = resp.json()
             file_id = resp_json.get("id")
             if file_id:
-                logging.info(f"File uploaded to OpenAI uploads: {filename}, file_id: {file_id}")
+                logging.info(f"File uploaded to OpenAI /v1/files: {filename}, file_id: {file_id}")
                 return file_id
             else:
-                logging.error(f"OpenAI /v1/uploads did not return file id. Response: {resp_json}")
+                logging.error(f"OpenAI /v1/files did not return file id. Response: {resp_json}")
                 return None
         else:
-            logging.error(f"Failed uploading file to OpenAI API /v1/uploads: {resp.status_code}, {resp.text}")
+            logging.error(f"Failed uploading file to OpenAI API /v1/files: {resp.status_code}, {resp.text}")
             return None
     except Exception as e:
-        logging.error(f"Exception during file upload to OpenAI /v1/uploads: {e}")
+        logging.error(f"Exception during file upload to OpenAI /v1/files: {e}")
         return None
 
 def openai_stream_generate(prompt, file_id=None):
@@ -106,8 +102,12 @@ def openai_stream_generate(prompt, file_id=None):
         messages = [{"role": "user", "content": prompt}]
         extra_kwargs = {}
         if file_id:
-            extra_kwargs["files"] = [file_id]
-            logging.info(f"Passing file_id to OpenAI ChatCompletion: {file_id}")
+            if isinstance(file_id, list):
+                extra_kwargs["files"] = file_id
+                logging.info(f"Passing file_id list to OpenAI ChatCompletion: {file_id}")
+            else:
+                extra_kwargs["files"] = [file_id]
+                logging.info(f"Passing file_id to OpenAI ChatCompletion: {file_id}")
         stream = openai.ChatCompletion.create(
             model=OPENAI_MODEL,
             messages=messages,