Update app.py

app.py

@@ -349,7 +349,7 @@ def generate_text(df, country, theme):
     {row_str}
     Topic: {theme}
     
-    Describe the adverse growth in the provided data year by year. Ensure the description follows the pattern in the example, highlighting whether the values increased or decreased each year.
+    Describe the adverse growth in the provided data year by year following the pattern in the example, highlighting whether the values increased or decreased each year.
     """
     prompt1 = f"""
     Here is an example of how to describe adverse growth data for a given country:
@@ -536,26 +536,31 @@ with gr.Blocks(theme='gradio/soft',js=js_func) as demo:
         
         
         b1 = gr.Button("Compare Data")
-        b2 = gr.Button("Extract text information")
+        b2 = gr.Button("Extract text information from PDFs")
         
         with gr.Row():
             with gr.Column():
                 sentiment_results_pdf1 = gr.HighlightedText(label="Sentiment Analysis - PDF 1")
-                country_1_dropdown = gr.Dropdown(label="Select Country from Excel File 1")
-                summarize_btn1_country = gr.Button("Summary for the selected country")
-                text_result_df1 = gr.Textbox(label="Sentence for excel file 1", lines=2)
-                summarize_btn1_country.click(fn=lambda country, theme: generate_text(stored_df1, country, theme),
-                             inputs=[country_1_dropdown, sheet],
-                             outputs=text_result_df1)
+
             with gr.Column():
                 sentiment_results_pdf2 = gr.HighlightedText(label="Sentiment Analysis - PDF 2")
-                country_2_dropdown = gr.Dropdown(label="Select Country from Excel File 2")
-                summarize_btn2_country = gr.Button("Summary for the selected country")
-                text_result_df2 = gr.Textbox(label="Sentence for excel file 2", lines=2)
-                summarize_btn2_country.click(fn=lambda country, theme: generate_text(stored_df2, country, theme),
-                             inputs=[country_2_dropdown, sheet],
-                             outputs=text_result_df2)
-    
+               
+        with gr.Row():
+            with gr.Group("Adverse growth trends"):
+                with gr.Column():
+                    country_1_dropdown = gr.Dropdown(label="Select Country from Excel File 1")
+                    summarize_btn1_country = gr.Button("Summary for the selected country")
+                    text_result_df1 = gr.Textbox(label="Sentence for excel file 1", lines=2)
+                    summarize_btn1_country.click(fn=lambda country, theme: generate_text(stored_df1, country, theme),
+                                 inputs=[country_1_dropdown, sheet],
+                                 outputs=text_result_df1)
+                with gr.Column():
+                    country_2_dropdown = gr.Dropdown(label="Select Country from Excel File 2")
+                    summarize_btn2_country = gr.Button("Summary for the selected country")
+                    text_result_df2 = gr.Textbox(label="Sentence for excel file 2", lines=2)
+                    summarize_btn2_country.click(fn=lambda country, theme: generate_text(stored_df2, country, theme),
+                                 inputs=[country_2_dropdown, sheet],
+                                 outputs=text_result_df2)
         # Button to extract text from PDFs and perform sentiment analysis
         b1.click(fn=process_and_compare, inputs=[file1, sheet, file2, sheet], outputs=[result,country_1_dropdown, country_2_dropdown])
         b2.click(fn=process_pdfs_and_analyze_sentiment, inputs=[file1, file2, sheet], outputs=[sentiment_results_pdf1, sentiment_results_pdf2])