Spaces:

sanjaymalladi
/

document-processor

Build error

sanjaymalladi commited on Dec 27, 2024

Commit

0ae087f

1 Parent(s): 77014ce

changes made

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ import numpy as np
 import joblib
 import base64
 from werkzeug.utils import secure_filename
 # HTML template with embedded JavaScript
 HTML_TEMPLATE = """
 <!DOCTYPE html>
@@ -333,26 +333,25 @@ def batch_process():
     files = request.files.getlist('files[]')
     user_id = request.form.get('user_id')
-    file_paths = []
-    for file in files:
-        if file.filename.endswith('.pdf'):
-            temp_path = f"temp_{file.filename}"
-            file.save(temp_path)
-            file_paths.append(temp_path)
-    try:
-        results = processor.process_batch(file_paths, user_id)
-    except Exception as e:
-        return jsonify({'error': str(e)}), 500
-    finally:
-        # Clean up temporary files
-        for path in file_paths:
-            try:
-                os.remove(path)
-            except:
-                pass
-    return jsonify(results)
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=True)

 import joblib
 import base64
 from werkzeug.utils import secure_filename
+import tempfile
 # HTML template with embedded JavaScript
 HTML_TEMPLATE = """
 <!DOCTYPE html>
     files = request.files.getlist('files[]')
     user_id = request.form.get('user_id')
+    # Create a temporary directory
+    with tempfile.TemporaryDirectory() as temp_dir:
+        file_paths = []
+        for file in files:
+            if file.filename.endswith('.pdf'):
+                # Create a secure filename
+                secure_name = secure_filename(file.filename)
+                # Create full path in temporary directory
+                temp_path = os.path.join(temp_dir, secure_name)
+                file.save(temp_path)
+                file_paths.append(temp_path)
+        try:
+            results = processor.process_batch(file_paths, user_id)
+        except Exception as e:
+            return jsonify({'error': str(e)}), 500
+        # No need to manually clean up - TemporaryDirectory does it automatically
+        return jsonify(results)
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=True)