app

app.py

@@ -145,46 +145,46 @@ 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)
-            
-            # Copy the file to the destination folder
-            shutil.copy(source_file_path, destination_file_path)
+    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)
         
-        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}")
+        # 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)
+    
+    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
-            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)
+        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)))