Update app.py

app.py

@@ -1,29 +1,31 @@
 import gradio as gr
 import zipfile
 import os
+from tempfile import TemporaryDirectory
 
-def extract_zip_file(zip_file):
-    # Save the uploaded ZIP file to a temporary path
-    zip_file_path = 'uploaded_zip.zip'
-    with open(zip_file_path, 'wb') as f:
-        f.write(zip_file.read())
+def extract_zip(file_obj):
+    with TemporaryDirectory() as tmpdirname:
+        # Save the uploaded zip file to the temporary directory
+        zip_path = os.path.join(tmpdirname, 'uploaded.zip')
+        with open(zip_path, 'wb') as f:
+            f.write(file_obj.read())
+        
+        # Extract the zip file
+        with zipfile.ZipFile(zip_path, 'r') as zip_ref:
+            zip_ref.extractall(tmpdirname)
+        
+        # Return the path to the extracted files
+        return f"Extracted to {tmpdirname}"
 
-    # Specify the directory to extract files
-    extract_dir = 'extracted_files'
-    os.makedirs(extract_dir, exist_ok=True)
-
-    # Extract files
-    with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
-        zip_ref.extractall(extract_dir)
-
-    return f"All files from {zip_file_path} have been extracted to {extract_dir}."
-
-# Create a gradio interface
+# Create the Gradio interface
 iface = gr.Interface(
-    fn=extract_zip_file,
-    inputs=gr.File(label="Upload ZIP File"),
-    outputs="text"
+    fn=extract_zip,
+    inputs=gr.inputs.File(type="file", label="Upload ZIP File"),
+    outputs="text",
+    title="ZIP File Extractor",
+    description="Upload a ZIP file and it will be extracted."
 )
 
-# Launch the interface
-iface.launch()
+if __name__ == "__main__":
+    iface.launch()
+