Update final.py

final.py

@@ -84,6 +84,11 @@ uploaded_file_name = None
 import os
 import shutil
 
+import os
+import shutil
+import chainlit as cl
+from chainlit.types import AskFileResponse
+
 @cl.on_chat_start
 async def start():
     global uploaded_file_name
@@ -95,27 +100,43 @@ async def start():
             content="Please upload a zip file to begin!", accept={"application/zip": [".zip"]}
         ).send()
 
-    zip_file = files[0]  # Assuming only one file is uploaded
+    zip_file: AskFileResponse = files[0]  # Assuming only one file is uploaded
     uploaded_file_name = zip_file.name
 
+    # Print out the attributes of the zip_file object for debugging
+    print(f"zip_file attributes: {dir(zip_file)}")
+
     # Get the CHAINLIT_USER_FILES_DIR from environment variables
     user_files_dir = os.environ.get('CHAINLIT_USER_FILES_DIR', '/tmp/chainlit_user_files')
 
     # Ensure the user files directory exists
     os.makedirs(user_files_dir, exist_ok=True)
 
-    # Define the destination path
-    dest_path = os.path.join(user_files_dir, zip_file.name)
+    # Save the uploaded file directly to the user files directory
+    file_path = os.path.join(user_files_dir, zip_file.name)
 
-    # Copy the file to the destination
-    shutil.copy(zip_file.path, dest_path)
+    # Attempt to read the file content and save it
+    try:
+        with open(file_path, "wb") as f:
+            f.write(zip_file.content)  # This may need to be adjusted based on the attributes of zip_file
+    except AttributeError as e:
+        print(f"AttributeError: {e}")
+        # If zip_file.content doesn't exist, try another method
+        try:
+            with open(zip_file.path, "rb") as src_file:
+                with open(file_path, "wb") as dest_file:
+                    shutil.copyfileobj(src_file, dest_file)
+        except Exception as e:
+            print(f"Error while copying file: {e}")
+            await cl.Message(content=f"Error while copying file: {e}").send()
+            return
 
     # Let the user know that the system is ready
     await cl.Message(content=f"`{zip_file.name}` uploaded successfully!").send()
 
     # Ask if the user wants to proceed with grading
     await cl.Message(content="Do you want to proceed with the grading? (yes/no)").send()
-
+    
 async def process_grading():
     global uploaded_file_name
     if uploaded_file_name: