Update app.py

app.py

@@ -202,10 +202,22 @@ def generate_text(df, country, theme):
     #for column in df.columns:
     #    if column != 'Country':
     #        df[column] = df[column].apply(lambda x: f"{x:.6f}%")
-    row = df[df['Country'] == country].iloc[0]
-
+    #row = df[df['Country'] == country].iloc[0]
+    def format_row_for_prompt(row):
+        # Create a formatted string with colons and percentages
+        formatted_row = []
+        for col, value in row.items():
+            if col != 'Country':  # Exclude 'Country' or format differently if needed
+                if isinstance(value, (int, float)):  # Add percentage sign for numeric values
+                    value_str = f"{value:.6f}%"
+                else:
+                    value_str = str(value)
+                formatted_row.append(f"{col}: {value_str}")
+        return "\n".join(formatted_row)
     # Convert the row to a string format for prompt
-    row_str = row.to_string(index=True)
+    row = df[df['Country'] == country].iloc[0]
+    row_str = format_row_for_prompt(row)
+    #row_str = row.to_string(index=True)
     
     simple_prompt = f"""
     Here is the data for {theme} in {country}:
@@ -217,19 +229,20 @@ def generate_text(df, country, theme):
     prompt = f"""
     Here is an example of how to describe adverse growth data for a given country:
     Country: Australia
-    Adverse 2020: -0.427975
-    Adverse 2021: -1.987167
-    Adverse 2022: -1.195906
-    Adverse Cumulative: -3.573762
-    The topic is GDP.
+    Adverse 2020: -0.43%
+    Adverse 2021: -1.99%
+    Adverse 2022: -1.20%
+    Cumulative: -3.57%
     
     Description:
-    In the adverse scenario, the growth for GDP in Australia decreased by -0.427975% in 2020, worsened further by -1.987167% in 2021, and slightly improved by -1.195906% in 2022, resulting in an adverse cumulative decrease of -3.573762%.
+    In the adverse scenario, the growth for GDP in Australia was -0.43% in 2020, worsened further to -1.99% in 2021, and slightly improved to -1.20% in 2022. The cumulative adverse growth is -3.57%.
     
     Now, using the following data for {theme} in {country}, describe the adverse growth:
     {row_str}
     The topic is {theme}.
-    Make sure to highlight changes compared to previous years, include cumulative results, and reflect the data accurately following the example.
+    - Highlight changes between the years.
+    - Accurately describe whether the values increase or decrease.
+    - Include the cumulative result, and reflect the data as it appears.
     """