Update app.py

app.py

@@ -282,14 +282,14 @@ def generate_page_report(df_pages, report_file_name):
         '(Primary and Secondary) for each page with a probability score greater than zero.'
     )
 
-    doc.add_heading("📊 Primary SDGs Bar Graph", level=3)
+    doc.add_heading("Primary SDGs Bar Graph", level=3)
     doc.add_paragraph(
         'This graph displays the most essential SDG the AI model associates with pages. The bars '
         'represent the percentage of pages most strongly aligned with each SDG. This offers insight into the dominant '
         'sustainable development theme within the document.'
     )
 
-    doc.add_heading("📈 Secondary SDGs Bar Graph", level=3)
+    doc.add_heading("Secondary SDGs Bar Graph", level=3)
     doc.add_paragraph(
         'This graph shows the second most relevant SDGs for pages. Although these SDGs are '
         'not the primary focus, the text has some relevance to these goals.'
@@ -334,14 +334,14 @@ def generate_sentence_report(df_sentences, report_file_name):
         '(Primary and Secondary) for each sentence with a probability score greater than zero.'
     )
 
-    doc.add_heading("📊 Primary SDGs Bar Graph", level=3)
+    doc.add_heading("Primary SDGs Bar Graph", level=3)
     doc.add_paragraph(
         'This graph displays the most essential SDG the AI model associates with sentences. The bars '
         'represent the percentage of sentences most strongly aligned with each SDG. This offers more profound insight '
         'into the dominant sustainable development theme within the document.'
     )
 
-    doc.add_heading("📈 Secondary SDGs Bar Graph", level=3)
+    doc.add_heading("Secondary SDGs Bar Graph", level=3)
     doc.add_paragraph(
         'This graph shows the second most relevant SDGs for sentences. Although these SDGs are not '
         'the primary focus, the text has some relevance to these goals.'
@@ -483,6 +483,7 @@ def launch_interface():
         )
 
         # Shared PDF file input for both analyses
+        gr.Markdown("## Upload PDF File")
         with gr.Row():
             file_input = gr.File(
                 label="📁 Upload PDF File for Analysis", file_types=[".pdf"]
@@ -492,7 +493,8 @@ def launch_interface():
         gr.Markdown(
             """
             ## PDF Text Extraction Mode  
-            Choose whether to analyze all pages or a specific range of pages. If you want to exclude certain pages from the analysis, select "Range of Pages" and specify the start and end pages.
+            Choose whether to analyze all pages or a specific range of pages. If you want to exclude certain pages from the analysis, select 
+            "Range of Pages" and specify the start and end pages. 
             """
         )
         with gr.Row():
@@ -503,7 +505,7 @@ def launch_interface():
             )
 
         with gr.Row():
-            start_page = gr.Number(value=1, label="🔢 Start Page", visible=False)
+            start_page = gr.Number(value=1, label="🔢 Start Page", visible=False, info="The cover page is page 1")
             end_page = gr.Number(value=1, label="🔢 End Page", visible=False)
 
         # Function to update visibility of start_page and end_page
@@ -531,27 +533,44 @@ def launch_interface():
                 It provides **high-level SDG mapping** of documents at the page level.
                 """
             )
+
+            with gr.Row():
+                page_button = gr.Button("🏃‍♂️ Run Page-Level Analysis")
+                reset_page_button = gr.Button("🔄 Reset Page-Level Analysis", elem_classes="reset-button")
+
             # Nested Tabs for Primary and Secondary SDGs
             with gr.Tabs():
                 with gr.TabItem("📊 Primary SDGs"):
-                    primary_page_plot = gr.Plot(label="📊 Primary SDGs [Page-Level]")
-
+                    with gr.Row():
+                        primary_page_plot = gr.Plot(label="📊 Primary SDGs [Page-Level]", scale=2)
+                        gr.Textbox(
+                            "When the analysis is completed, the Primary SDGs bar graph on the left will show "+
+                            "the percentage of pages that strongly align with each SDG. The icon for the most frequent "+
+                            "SDG is highlighted above the graph. Download the Page Predictions CVS for further details.",
+                            interactive=False, scale=1
+                        )
+                    gr.Markdown("##### Download results")    
                     with gr.Row():
                         page_csv = gr.File(label="📊 Download Page Predictions CSV")
                         page_docx = gr.File(label="📄 Download Page Report DOCX")
                         page_jpeg1 = gr.File(label="🖼️ Download Primary SDGs JPEG")
 
                 with gr.TabItem("📈 Secondary SDGs"):
+                    with gr.Row():
+                        primary_page_plot = gr.Plot(label="📊 Primary SDGs [Page-Level]", scale=2)
+                        gr.Textbox(
+                            "When the analysis is completed, the Secondary SDGs bar graph on the left will show "+
+                            "SDGs that are not the primary focus of the pages analysed. These SDGs are second to the "+
+                            "Primary SDGs.",
+                            interactive=False, scale=1
+                        )
+                    gr.Markdown("##### Download results") 
                     secondary_page_plot = gr.Plot(label="📈 Secondary SDGs [Page-Level]")
 
                     with gr.Row():
                         page_csv_secondary = gr.File(label="📊 Download Page Predictions CSV")
                         page_report_file_secondary = gr.File(label="📄 Download Page Report DOCX")
-                        secondary_page_jpeg = gr.File(label="🖼️ Download Secondary SDGs JPEG")
-
-            with gr.Row():
-                page_button = gr.Button("🏃‍♂️ Run Page-Level Analysis")
-                reset_page_button = gr.Button("🔄 Reset Page-Level Analysis", elem_classes="reset-button")
+                        secondary_page_jpeg = gr.File(label="🖼️ Download Secondary SDGs JPEG")    
 
         with gr.Tab("✍️ Sentence-Level Analysis"):
             gr.Markdown(
@@ -562,11 +581,23 @@ def launch_interface():
                 It provides **detailed SDG mapping** at the sentence level.
                 """
             )
+
+            with gr.Row():
+                sentence_button = gr.Button("🏃‍♂️ Run Sentence-Level Analysis")
+                reset_sentence_button = gr.Button("🔄 Reset Sentence-Level Analysis", elem_classes="reset-button")
+
             # Nested Tabs for Primary and Secondary SDGs
             with gr.Tabs():
                 with gr.TabItem("📊 Primary SDGs"):
-                    primary_sentence_plot = gr.Plot(label="📊 Primary SDGs [Sentence-Level]")
-
+                    with gr.Row():
+                        primary_sentence_plot = gr.Plot(label="📊 Primary SDGs [Sentence-Level]", scale=2)
+                        gr.Textbox(
+                            "When the analysis is completed, the Primary SDGs bar graph on the left will show "+
+                            "the percentage of sentences that strongly align with each SDG. The icon for the most frequent "+
+                            "SDG is highlighted above the graph. Download the Sentence Predictions CVS for further details.",
+                            interactive=False, scale=1
+                        )
+                    gr.Markdown("##### Download results")  
                     with gr.Row():
                         sentence_csv = gr.File(label="📊 Download Sentence Predictions CSV")
                         sentence_docx = gr.File(label="📄 Download Sentence Report DOCX")
@@ -579,11 +610,7 @@ def launch_interface():
                         sentence_csv_secondary = gr.File(label="📊 Download Sentence Predictions CSV")
                         sentence_report_file_secondary = gr.File(label="📄 Download Sentence Report DOCX")
                         secondary_sentence_jpeg = gr.File(label="🖼️ Download Secondary SDGs JPEG")
-
-            with gr.Row():
-                sentence_button = gr.Button("🏃‍♂️ Run Sentence-Level Analysis")
-                reset_sentence_button = gr.Button("🔄 Reset Sentence-Level Analysis", elem_classes="reset-button")
-
+            
         # Function to process page-level analysis
         @spaces.GPU
         def process_pages(file, extraction_mode, start_page, end_page):
@@ -633,13 +660,13 @@ def launch_interface():
                 )
 
                 # Define output file names
-                page_csv_file = f"{sanitized_file_name}_page_predictions_primary.csv"
-                page_report_file = f"{sanitized_file_name}_page_report_primary.docx"
-                primary_page_jpeg = f"{sanitized_file_name}_primary_page.jpeg"
+                page_csv_file = f"{sanitized_file_name}_SDG-Page_predictions.csv"
+                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}_page_predictions_secondary.csv"
-                page_report_file_secondary = f"{sanitized_file_name}_page_report_secondary.docx"
-                secondary_page_jpeg = f"{sanitized_file_name}_secondary_page.jpeg"
+                page_csv_file_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"
 
                 # Save CSV and reports
                 df_page_predictions.to_csv(page_csv_file, index=False)
@@ -711,13 +738,13 @@ def launch_interface():
                 )
 
                 # Define output file names
-                sentence_csv_file = f"{sanitized_file_name}_sentence_predictions_primary.csv"
-                sentence_report_file = f"{sanitized_file_name}_sentence_report_primary.docx"
-                primary_sentence_jpeg = f"{sanitized_file_name}_primary_sentence.jpeg"
+                sentence_csv_file = f"{sanitized_file_name}_SDG-Sentence_predictions.csv"
+                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}_sentence_predictions_secondary.csv"
-                sentence_report_file_secondary = f"{sanitized_file_name}_sentence_report_secondary.docx"
-                secondary_sentence_jpeg = f"{sanitized_file_name}_secondary_sentence.jpeg"
+                sentence_csv_file_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"
 
                 # Save CSV and reports
                 df_sentence_predictions.to_csv(sentence_csv_file, index=False)
@@ -757,8 +784,8 @@ def launch_interface():
                 page_csv,                    # 📊 Download Page Predictions CSV
                 page_docx,                   # 📄 Download Page Report DOCX
                 page_jpeg1,                  # 🖼️ Download Primary SDGs JPEG
-                page_csv_secondary,          # 📊 Download Page Predictions CSV (Secondary)
-                page_report_file_secondary,  # 📄 Download Page Report DOCX (Secondary)
+                page_csv_secondary,          # 📊 Download Page Predictions CSV
+                page_report_file_secondary,  # 📄 Download Page Report DOCX
                 secondary_page_jpeg          # 🖼️ Download Secondary SDGs JPEG
             ]
         )
@@ -787,8 +814,8 @@ def launch_interface():
                 sentence_csv,                    # 📊 Download Sentence Predictions CSV
                 sentence_docx,                   # 📄 Download Sentence Report DOCX
                 sentence_jpeg1,                  # 🖼️ Download Primary SDGs JPEG
-                sentence_csv_secondary,          # 📊 Download Sentence Predictions CSV (Secondary)
-                sentence_report_file_secondary,  # 📄 Download Sentence Report DOCX (Secondary)
+                sentence_csv_secondary,          # 📊 Download Sentence Predictions CSV
+                sentence_report_file_secondary,  # 📄 Download Sentence Report DOCX
                 secondary_sentence_jpeg          # 🖼️ Download Secondary SDGs JPEG
             ]
         )