app

app.py

@@ -145,48 +145,47 @@ def predict_files(filenames):
     image_paths = eval(filenames)  # Convert the filenames string back to a list
     
     for filename in image_paths:
-    file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
-    folder_path = UPLOAD_FOLDER
-    destination_folder = r'static/temp/img_display'
-    if not os.path.exists(destination_folder):
-        os.makedirs(destination_folder)
-
-    # Get a list of all files in the source folder
-    files = os.listdir(folder_path)
-
-    # Loop through each file and copy it to the destination folder
-    for file in files:
-        # Construct the full path of the source file
-        source_file_path = os.path.join(folder_path, file)
-        
-        # Construct the full path of the destination file
-        destination_file_path = os.path.join(destination_folder, file)
+        file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
+        folder_path = UPLOAD_FOLDER
+        destination_folder = r'static/temp/img_display'
+        if not os.path.exists(destination_folder):
+            os.makedirs(destination_folder)
+
+        # Get a list of all files in the source folder
+        files = os.listdir(folder_path)
+
+        # Loop through each file and copy it to the destination folder
+        for file in files:
+            # Construct the full path of the source file
+            source_file_path = os.path.join(folder_path, file)
+            
+            # Construct the full path of the destination file
+            destination_file_path = os.path.join(destination_folder, file)
+            
+            # Copy the file to the destination folder
+            shutil.copy(source_file_path, destination_file_path)
         
-        # Copy the file to the destination folder
-        shutil.copy(source_file_path, destination_file_path)
-    
-    if os.path.exists(file_path):
-        # Call make_predictions automatically
-        prediction_result = make_predictions([file_path])
-        if isinstance(prediction_result, list) and len(prediction_result) > 0:
-            prediction_results.append(prediction_result[0])  # Append only the first prediction result
-        else:
-            print(f"Error making prediction for {file}: {prediction_result}")
-
-        prediction_results_copy = copy.deepcopy(prediction_results)
-
-        non_receipt_indices = []
-        for i, prediction in enumerate(prediction_results):
-            if prediction == 'non-receipt':
-                non_receipt_indices.append(i)
-
-        # Delete images in reverse order to avoid index shifting
-        for index in non_receipt_indices[::-1]:
-            file_to_remove = os.path.join('static', 'temp', 'uploads', image_paths[index])
-            if os.path.exists(file_to_remove):
-                os.remove(file_to_remove)
+        if os.path.exists(file_path):
+            # Call make_predictions automatically
+            prediction_result = make_predictions([file_path])
+            if isinstance(prediction_result, list) and len(prediction_result) > 0:
+                prediction_results.append(prediction_result[0])  # Append only the first prediction result
+            else:
+                print(f"Error making prediction for {file}: {prediction_result}")
+
+            prediction_results_copy = copy.deepcopy(prediction_results)
+
+            non_receipt_indices = []
+            for i, prediction in enumerate(prediction_results):
+                if prediction == 'non-receipt':
+                    non_receipt_indices.append(i)
+
+            # Delete images in reverse order to avoid index shifting
+            for index in non_receipt_indices[::-1]:
+                file_to_remove = os.path.join('static', 'temp', 'uploads', image_paths[index])
+                if os.path.exists(file_to_remove):
+                    os.remove(file_to_remove)
                     
-
     return render_template('extractor.html', index_url=index_url, image_paths=image_paths, prediction_results = prediction_results, predictions=dict(zip(image_paths, prediction_results_copy)))