WebashalarForML commited on
Commit
a321f36
·
verified ·
1 Parent(s): a170fd5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -12,7 +12,7 @@ from waitress import serve
12
  # Initialize the Flask application
13
  app = Flask(__name__)
14
  app.secret_key = 'your_secret_key'
15
- app.config['UPLOAD_FOLDER'] = 'uploads'
16
 
17
  # Allowed file extensions
18
  ALLOWED_EXTENSIONS = {'pdf', 'docx', 'rsf', 'odt', 'png', 'jpg', 'jpeg'}
@@ -51,6 +51,11 @@ def upload_and_process():
51
  logging.debug(f"File uploaded: {filename}")
52
  session['uploaded_file'] = filename
53
 
 
 
 
 
 
54
  # Process the file after uploading
55
  try:
56
  parsed_data = process_resume_data(file_path)
@@ -101,10 +106,12 @@ def reset_upload():
101
  def result():
102
  """Display the processed data result."""
103
  processed_data = session.get('processed_data', None)
 
 
104
  if not processed_data:
105
  flash('No data to display. Please upload and process a file.')
106
  return redirect(url_for('index'))
107
- return render_template('result.html', parsed_data=processed_data)
108
 
109
  if __name__ == '__main__':
110
  # For Windows development
 
12
  # Initialize the Flask application
13
  app = Flask(__name__)
14
  app.secret_key = 'your_secret_key'
15
+ app.config['UPLOAD_FOLDER'] = 'uploads/'
16
 
17
  # Allowed file extensions
18
  ALLOWED_EXTENSIONS = {'pdf', 'docx', 'rsf', 'odt', 'png', 'jpg', 'jpeg'}
 
51
  logging.debug(f"File uploaded: {filename}")
52
  session['uploaded_file'] = filename
53
 
54
+ file_url = f"/uploads/{filename}"
55
+ file_extension = filename.rsplit('.', 1)[1].lower()
56
+ session['file_url'] = file_url
57
+ session['file_extension'] = file_extension
58
+
59
  # Process the file after uploading
60
  try:
61
  parsed_data = process_resume_data(file_path)
 
106
  def result():
107
  """Display the processed data result."""
108
  processed_data = session.get('processed_data', None)
109
+ file_url = session.get('file_url', None)
110
+ file_extension = session.get('file_extension', None)
111
  if not processed_data:
112
  flash('No data to display. Please upload and process a file.')
113
  return redirect(url_for('index'))
114
+ return render_template('result.html', parsed_data=processed_data, file_url=file_url, file_extension=file_extension)
115
 
116
  if __name__ == '__main__':
117
  # For Windows development