Update app.py

app.py

@@ -86,7 +86,7 @@ def remove_file():
 
 @app.route('/process', methods=['POST'])
 def process_file():
-    uploaded_files = session.get('uploaded_files', [])  # Get files from the session
+    uploaded_files = session.get('uploaded_files', [])
     if not uploaded_files:
         flash('No files selected for processing')
         logging.warning("No files selected for processing")
@@ -97,6 +97,8 @@ def process_file():
     logging.info(f"Processing files: {file_paths}")
 
     extracted_text = {}  # Initialize extracted_text
+    processed_Img = {}  # Initialize processed_Img to avoid the UnboundLocalError
+
     try:
         # Extract text from all images
         extracted_text, processed_Img = extract_text_from_images(file_paths, RESULT_FOLDER)
@@ -114,24 +116,26 @@ def process_file():
         logging.error(f"Error during LLM processing: {e}")
         logging.info("Running backup model...")
 
-        # Default assignment for LLMdata in case of error
-        LLMdata = {}
-        # Run the backup model in case of an exception
+        LLMdata = {}  # Ensure LLMdata is initialized
+       
         if extracted_text:
             text = json_to_llm_str(extracted_text)
             LLMdata = NER_Model(text)
             logging.info(f"NER model data: {LLMdata}")
         else:
             logging.warning("No extracted text available for backup model")
-    
+
+    # Processing results
     cont_data = process_extracted_text(extracted_text)
     logging.info(f"Contextual data: {cont_data}")
     
-    # Storing the parsed results in session
+    # Storing the parsed results
     processed_data = process_resume_data(LLMdata, cont_data, extracted_text)
+    logging.info(f"Processed data: {processed_data}")
+
+    # Store processed data and images in session
     session['processed_data'] = processed_data
-    session['processed_Img'] = processed_Img
-    session.modified = True  # Ensure session is updated
+    session['processed_Img'] = processed_Img  # This will be either empty or populated from the try block
     flash('Data processed and analyzed successfully')
     logging.info("Data processed and analyzed successfully")
     return redirect(url_for('result'))