Update app.py

app.py

@@ -1,115 +1,116 @@
-from flask import Flask, request, redirect, flash, session, render_template, url_for
-import os
-import json
-import logging
-from werkzeug.utils import secure_filename
-from utils.error import handle_file_not_found, handle_invalid_file_type, handle_file_processing_error, page_not_found, internal_server_error
-from utils.spacy import Parser_from_model
-from utils.mistral import process_resume_data
-import platform
-from waitress import serve
-
-# Initialize the Flask application
-app = Flask(__name__)
-app.secret_key = 'your_secret_key'
-app.config['UPLOAD_FOLDER'] = 'uploads'
-
-# Allowed file extensions
-ALLOWED_EXTENSIONS = {'pdf', 'docx', 'rsf', 'odt', 'png', 'jpg', 'jpeg'}
-
-# Configure logging
-logging.basicConfig(level=logging.DEBUG)
-
-# Error handlers
-app.register_error_handler(404, page_not_found)
-app.register_error_handler(500, internal_server_error)
-
-def allowed_file(filename):
-    """Check if the file has an allowed extension."""
-    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
-
-@app.route('/')
-def index():
-    """Display the index page with the uploaded file information."""
-    uploaded_file = session.get('uploaded_file', None)
-    return render_template('index.html', uploaded_file=uploaded_file)
-
-@app.route('/upload_and_process', methods=['POST'])
-def upload_and_process():
-    """Handle file upload and process the file."""
-    if 'file' not in request.files or request.files['file'].filename == '':
-        flash('No file selected for upload.')
-        return redirect(request.url)
-
-    file = request.files['file']
-    
-    # Check if the file is allowed
-    if file and allowed_file(file.filename):
-        filename = secure_filename(file.filename)
-        file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
-        file.save(file_path)
-        logging.debug(f"File uploaded: {filename}")
-        session['uploaded_file'] = filename
-
-        # Process the file after uploading
-        try:
-            parsed_data = process_resume_data(file_path)
-            if not parsed_data or 'error' in parsed_data:
-                flash('An error occurred during file processing.')
-                return redirect(url_for('index'))
-
-            session['processed_data'] = parsed_data
-            flash('File uploaded and data processed successfully.')
-            return redirect(url_for('result'))
-
-        except Exception as e:
-            logging.error(f"File processing error: {str(e)}")
-            flash('An error occurred while processing the file.')
-            return handle_file_processing_error()
-    else:
-        return handle_invalid_file_type()
-
-@app.route('/remove_file', methods=['POST'])
-def remove_file():
-    """Remove the uploaded file and reset the session."""
-    uploaded_file = session.get('uploaded_file')
-    if uploaded_file:
-        file_path = os.path.join(app.config['UPLOAD_FOLDER'], uploaded_file)
-        if os.path.exists(file_path):
-            os.remove(file_path)
-        session.pop('uploaded_file', None)
-        flash('File successfully removed.')
-    else:
-        flash('No file to remove.')
-    return redirect(url_for('index'))
-
-@app.route('/reset_upload')
-def reset_upload():
-    """Reset the uploaded file and the processed data."""
-    uploaded_file = session.get('uploaded_file')
-    if uploaded_file:
-        file_path = os.path.join(app.config['UPLOAD_FOLDER'], uploaded_file)
-        if os.path.exists(file_path):
-            os.remove(file_path)
-        session.pop('uploaded_file', None)
-
-    session.pop('processed_data', None)
-    flash('File and data reset. You can upload a new file.')
-    return redirect(url_for('index'))
-
-@app.route('/result')
-def result():
-    """Display the processed data result."""
-    processed_data = session.get('processed_data', None)
-    if not processed_data:
-        flash('No data to display. Please upload and process a file.')
-        return redirect(url_for('index'))
-    return render_template('result.html', parsed_data=processed_data)
-
-if __name__ == '__main__':
-    # For Windows development
-    if platform.system() == "Windows":
-        app.run(debug=True)
-    # For Linux or production with Waitress
-    else:
-        serve(app, host="0.0.0.0", port=7860)
+from flask import Flask, request, redirect, flash, session, render_template, url_for
+import os
+import json
+import logging
+from werkzeug.utils import secure_filename
+from utils.error import handle_file_not_found, handle_invalid_file_type, handle_file_processing_error, page_not_found, internal_server_error
+from utils.spacy import Parser_from_model
+from utils.mistral import process_resume_data
+import platform
+from waitress import serve
+
+# Initialize the Flask application
+app = Flask(__name__)
+app.secret_key = 'your_secret_key'
+app.config['UPLOAD_FOLDER'] = 'uploads'
+
+# Allowed file extensions
+ALLOWED_EXTENSIONS = {'pdf', 'docx', 'rsf', 'odt', 'png', 'jpg', 'jpeg'}
+
+# Configure logging
+logging.basicConfig(level=logging.DEBUG)
+
+# Error handlers
+app.register_error_handler(404, page_not_found)
+app.register_error_handler(500, internal_server_error)
+
+def allowed_file(filename):
+    """Check if the file has an allowed extension."""
+    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+
+@app.route('/')
+def index():
+    session.clear()
+    """Display the index page with the uploaded file information."""
+    uploaded_file = session.get('uploaded_file', None)
+    return render_template('index.html', uploaded_file=uploaded_file)
+
+@app.route('/upload_and_process', methods=['POST'])
+def upload_and_process():
+    """Handle file upload and process the file."""
+    if 'file' not in request.files or request.files['file'].filename == '':
+        flash('No file selected for upload.')
+        return redirect(request.url)
+
+    file = request.files['file']
+    
+    # Check if the file is allowed
+    if file and allowed_file(file.filename):
+        filename = secure_filename(file.filename)
+        file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
+        file.save(file_path)
+        logging.debug(f"File uploaded: {filename}")
+        session['uploaded_file'] = filename
+
+        # Process the file after uploading
+        try:
+            parsed_data = process_resume_data(file_path)
+            if not parsed_data or 'error' in parsed_data:
+                flash('An error occurred during file processing.')
+                return redirect(url_for('index'))
+
+            session['processed_data'] = parsed_data
+            flash('File uploaded and data processed successfully.')
+            return redirect(url_for('result'))
+
+        except Exception as e:
+            logging.error(f"File processing error: {str(e)}")
+            flash('An error occurred while processing the file.')
+            return handle_file_processing_error()
+    else:
+        return handle_invalid_file_type()
+
+@app.route('/remove_file', methods=['POST'])
+def remove_file():
+    """Remove the uploaded file and reset the session."""
+    uploaded_file = session.get('uploaded_file')
+    if uploaded_file:
+        file_path = os.path.join(app.config['UPLOAD_FOLDER'], uploaded_file)
+        if os.path.exists(file_path):
+            os.remove(file_path)
+        session.pop('uploaded_file', None)
+        flash('File successfully removed.')
+    else:
+        flash('No file to remove.')
+    return redirect(url_for('index'))
+
+@app.route('/reset_upload')
+def reset_upload():
+    """Reset the uploaded file and the processed data."""
+    uploaded_file = session.get('uploaded_file')
+    if uploaded_file:
+        file_path = os.path.join(app.config['UPLOAD_FOLDER'], uploaded_file)
+        if os.path.exists(file_path):
+            os.remove(file_path)
+        session.pop('uploaded_file', None)
+
+    session.pop('processed_data', None)
+    flash('File and data reset. You can upload a new file.')
+    return redirect(url_for('index'))
+
+@app.route('/result')
+def result():
+    """Display the processed data result."""
+    processed_data = session.get('processed_data', None)
+    if not processed_data:
+        flash('No data to display. Please upload and process a file.')
+        return redirect(url_for('index'))
+    return render_template('result.html', parsed_data=processed_data)
+
+if __name__ == '__main__':
+    # For Windows development
+    if platform.system() == "Windows":
+        app.run(debug=True)
+    # For Linux or production with Waitress
+    else:
+        serve(app, host="0.0.0.0", port=7860)