Update app.py

app.py

@@ -112,7 +112,7 @@ def list_uploaded_files():
         
         return jsonify({"files": files}), 200
     except Exception as e:
-        return jsonify({"error": f"Error listing files: {e}"}), 500
+        return jsonify({"error": f"Error listing files: {e}"}), 504
 
 @app.route("/upload", methods=["POST"])
 def handle_upload():
@@ -131,19 +131,19 @@ def handle_upload():
         if not os.access(UPLOAD_DIR, os.W_OK):  # Check write permissions
             os.chmod(UPLOAD_DIR, 0o777)
     except PermissionError as e:
-        return jsonify({"error": f"Permission error with upload directory: {e}"}), 500
+        return jsonify({"error": f"Permission error with upload directory: {e}"}), 501
 
     try:
         # Save the file to the upload directory
         file.save(file_path)
     except Exception as e:
-        return jsonify({"error": f"Error saving file: {e}"}), 500
+        return jsonify({"error": f"Error saving file: {e}"}), 502
     
     # Process the document using the upload_document function
     try:
         upload_document(file_path, bertmodel)  # Assuming 'bertmodel' is defined elsewhere
     except Exception as e:
-        return jsonify({"error": f"Error processing file: {e}"}), 500
+        return jsonify({"error": f"Error processing file: {e}"}), 503
     
     # Return success response
     return jsonify({"message": "File uploaded and processed successfully"}), 200