Update app.py

app.py

@@ -631,7 +631,7 @@ def launch_interface():
         def process_pages(file, extraction_mode, start_page, end_page):
             if not file:
                 # Return None for each output component
-                return [None, None, None, None, None, None, None, None]
+                return [None] * 8
 
             try:
                 if hasattr(file, 'name'):
@@ -687,7 +687,7 @@ def launch_interface():
                 page_report_file = f"{sanitized_file_name}_SDG-Page_report.docx"
                 primary_page_jpeg = f"{sanitized_file_name}_SDG-Page_primary_graph.jpeg"
 
-                page_csv_file_secondary = f"{sanitized_file_name}_SDG-Page_predictions.csv"
+                page_csv_secondary = f"{sanitized_file_name}_SDG-Page_predictions.csv"
                 page_report_file_secondary = f"{sanitized_file_name}_SDG-Page_report.docx"
                 secondary_page_jpeg = f"{sanitized_file_name}_SDG-Page_secondary_graph.jpeg"
 
@@ -708,21 +708,21 @@ def launch_interface():
                     page_csv_file, 
                     page_report_file, 
                     primary_page_jpeg,
-                    page_csv_file_secondary, 
+                    page_csv_secondary, 
                     page_report_file_secondary, 
                     secondary_page_jpeg
                 )
 
             except Exception as e:
                 print(f"Error: {e}")
-                return [None, None, None, None, None, None, None, None]
+                return [None] * 8
 
         # Function to process sentence-level analysis
         @spaces.GPU
         def process_sentences(file, extraction_mode, start_page, end_page):
             if not file:
                 # Return None for each output component
-                return [None, None, None, None, None, None, None, None]
+                return [None] * 8
 
             try:
                 if hasattr(file, 'name'):
@@ -778,7 +778,7 @@ def launch_interface():
                 sentence_report_file = f"{sanitized_file_name}_SDG-Sentence_report.docx"
                 primary_sentence_jpeg = f"{sanitized_file_name}_SDG-Sentence_primary_graph.jpeg"
 
-                sentence_csv_file_secondary = f"{sanitized_file_name}_SDG-Sentence_predictions.csv"
+                sentence_csv_secondary = f"{sanitized_file_name}_SDG-Sentence_predictions.csv"
                 sentence_report_file_secondary = f"{sanitized_file_name}_SDG-Sentence_report.docx"
                 secondary_sentence_jpeg = f"{sanitized_file_name}_SDG-Sentence_secondary_graph.jpeg"
 
@@ -799,14 +799,14 @@ def launch_interface():
                     sentence_csv_file, 
                     sentence_report_file, 
                     primary_sentence_jpeg,
-                    sentence_csv_file_secondary, 
+                    sentence_csv_secondary, 
                     sentence_report_file_secondary, 
                     secondary_sentence_jpeg
                 )
 
             except Exception as e:
                 print(f"Error: {e}")
-                return [None, None, None, None, None, None, None, None]
+                return [None] * 8
 
         # Reset functions to clear the outputs
         def reset_page_outputs():
@@ -825,7 +825,7 @@ def launch_interface():
                 page_csv,                    # 📊 Download Page Predictions CSV
                 page_docx,                   # 📄 Download Page Report DOCX
                 page_jpeg1,                  # 🖼️ Download Primary SDGs JPEG
-                page_csv_file_secondary,     # 📊 Download Page Predictions CSV
+                page_csv_secondary,          # 📊 Download Page Predictions CSV
                 page_report_file_secondary,  # 📄 Download Page Report DOCX
                 secondary_page_jpeg          # 🖼️ Download Secondary SDGs JPEG
             ]
@@ -839,7 +839,7 @@ def launch_interface():
                 page_csv,
                 page_docx,
                 page_jpeg1,
-                page_csv_file_secondary,
+                page_csv_secondary,
                 page_report_file_secondary,
                 secondary_page_jpeg
             ]
@@ -855,7 +855,7 @@ def launch_interface():
                 sentence_csv,                    # 📊 Download Sentence Predictions CSV
                 sentence_docx,                   # 📄 Download Sentence Report DOCX
                 sentence_jpeg1,                  # 🖼️ Download Primary SDGs JPEG
-                sentence_csv_file_secondary,     # 📊 Download Sentence Predictions CSV
+                sentence_csv_secondary,          # 📊 Download Sentence Predictions CSV
                 sentence_report_file_secondary,  # 📄 Download Sentence Report DOCX
                 secondary_sentence_jpeg          # 🖼️ Download Secondary SDGs JPEG
             ]
@@ -869,7 +869,7 @@ def launch_interface():
                 sentence_csv,
                 sentence_docx,
                 sentence_jpeg1,
-                sentence_csv_file_secondary,
+                sentence_csv_secondary,
                 sentence_report_file_secondary,
                 secondary_sentence_jpeg
             ]