Update final.py

final.py

@@ -81,6 +81,9 @@ def process_data(zip_file_name: str, prompt_template) -> Tuple[float, float, int
 user_wants_to_continue = False
 uploaded_file_name = None
 
+import os
+import shutil
+
 @cl.on_chat_start
 async def start():
     global uploaded_file_name
@@ -92,22 +95,20 @@ async def start():
             content="Please upload a zip file to begin!", accept={"application/zip": [".zip"]}
         ).send()
 
-    zip_file: AskFileResponse = files[0]  # Assuming only one file is uploaded
+    zip_file = files[0]  # Assuming only one file is uploaded
     uploaded_file_name = zip_file.name
 
     # Get the CHAINLIT_USER_FILES_DIR from environment variables
     user_files_dir = os.environ.get('CHAINLIT_USER_FILES_DIR', '/tmp/chainlit_user_files')
 
-    # Create a temporary directory to store the uploaded file
-    with tempfile.TemporaryDirectory() as temp_dir:
-        temp_file_path = os.path.join(temp_dir, zip_file.name)
-        
-        # Copy the uploaded file to the temporary directory
-        shutil.copy(zip_file.path, temp_file_path)
-        
-        # Move the file to the user files directory
-        os.makedirs(user_files_dir, exist_ok=True)
-        shutil.move(temp_file_path, os.path.join(user_files_dir, zip_file.name))
+    # 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)
+
+    # Copy the file to the destination
+    shutil.copy(zip_file.path, dest_path)
 
     # Let the user know that the system is ready
     await cl.Message(content=f"`{zip_file.name}` uploaded successfully!").send()