File size: 6,198 Bytes
be28faf |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 |
from flask import Flask, request, redirect, flash, session, render_template, url_for
import os
import json
from werkzeug.utils import secure_filename
import logging
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
if platform.system() == "Windows":
app = Flask(__name__)
app.secret_key = 'your_secret_key'
app.config['UPLOAD_FOLDER'] = 'uploads'
# else:
# # For Hugging Face Spaces or other Linux environments
# if __name__ != "__main__":
# serve(app, host="0.0.0.0", port=7860)
# Error handlers
app.register_error_handler(404, page_not_found)
app.register_error_handler(500, internal_server_error)
# Allowed extensions
ALLOWED_EXTENSIONS = {'pdf', 'docx', 'rsf', 'odt', 'png', 'jpg', 'jpeg'}
def allowed_file(filename):
return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
@app.route('/')
def index():
uploaded_file = session.get('uploaded_file', None)
return render_template('index.html', uploaded_file=uploaded_file)
# @app.route('/upload', methods=['POST'])
# def upload_file():
# if 'file' not in request.files:
# flash('No file part')
# return redirect(request.url)
# file = request.files['file']
# if file.filename == '':
# flash('No selected file')
# return redirect(request.url)
# if file and allowed_file(file.filename):
# filename = secure_filename(file.filename)
# file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
# logging.debug(f"File uploaded: {filename}")
# session['uploaded_file'] = filename
# flash('File successfully uploaded')
# return redirect(url_for('index'))
# else:
# return handle_invalid_file_type()
# def process_file():
# selected_file = session.get('uploaded_file')
# if not selected_file:
# flash('No file selected for processing')
# return redirect(url_for('index'))
# file_path = os.path.join(app.config['UPLOAD_FOLDER'], selected_file)
# if not os.path.exists(file_path):
# return handle_file_not_found()
# parsed_data = process_resume_data(file_path)
# if not parsed_data or 'error' in parsed_data:
# return handle_file_processing_error()
# session['processed_data'] = parsed_data
# flash('Data processed successfully')
# return redirect(url_for('result'))
@app.route('/upload_and_process', methods=['POST', 'GET'])
def upload_and_process():
if 'file' not in request.files:
flash('No file part')
return redirect(request.url)
file = request.files['file']
if file.filename == '':
flash('No selected file')
return redirect(request.url)
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
parsed_data = process_resume_data(file_path)
if not parsed_data or 'error' in parsed_data:
return handle_file_processing_error()
session['processed_data'] = parsed_data
flash('File uploaded and data processed successfully')
return redirect(url_for('result'))
else:
return handle_invalid_file_type()
@app.route('/remove_file')
def remove_file():
uploaded_file = session.get('uploaded_file')
if uploaded_file:
os.remove(os.path.join(app.config['UPLOAD_FOLDER'], uploaded_file))
session.pop('uploaded_file', None)
flash('File successfully removed')
return redirect(url_for('index'))
@app.route('/reset_upload')
def reset_upload():
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 reset. You can upload a new file.')
return redirect(url_for('index'))
# @app.route('/process', methods=['GET', 'POST'])
# def process_file():
# selected_file = session.get('uploaded_file')
# if not selected_file:
# flash('No file selected for processing')
# return redirect(url_for('index'))
# file_path = os.path.join(app.config['UPLOAD_FOLDER'], selected_file)
# if not os.path.exists(file_path):
# return handle_file_not_found()
# parsed_data = process_resume_data(file_path)
# if not parsed_data or 'error' in parsed_data:
# return handle_file_processing_error()
# session['processed_data'] = parsed_data
# flash('Data processed successfully')
# return redirect(url_for('result'))
@app.route('/loading')
def loading():
selected_file = session.get('uploaded_file')
if not selected_file:
flash('No file selected for processing')
return redirect(url_for('index'))
file_path = os.path.join(app.config['UPLOAD_FOLDER'], selected_file)
parsed_data = process_resume_data(file_path)
if parsed_data and 'error' not in parsed_data:
session['processed_data'] = json.loads(parsed_data)
flash('Data processed successfully')
return redirect(url_for('result'))
else:
return handle_file_processing_error()
@app.route('/result')
def 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__':
app.run(debug=True)
|