Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 import yfinance as yf
 import pandas as pd
-import random
 
 # Mapping company names to their ticker symbols
 company_dict = {
@@ -34,12 +33,16 @@ def fetch_esg_data(company_name):
             "Score": esg_scores.squeeze().values
         })
         
-        return plot_df
+        # Save the ESG data to a CSV file
+        csv_filename = f"{ticker}_esg_data.csv"
+        esg_df.to_csv(csv_filename)
+
+        return plot_df, csv_filename  # Return the plot DataFrame and the CSV filename
     else:
-        # Return an empty DataFrame if no data is available
-        return pd.DataFrame()
+        # Return an empty DataFrame and None if no data is available
+        return pd.DataFrame(), None
 
-# Gradio interface with a dropdown for company selection and line plot visualization
+# Gradio interface with a dropdown for company selection, line plot visualization, and CSV download
 def app_interface():
     with gr.Blocks() as app:
         # Dropdown to select company name
@@ -51,10 +54,16 @@ def app_interface():
         # LinePlot component for displaying the ESG data
         plot_output = gr.LinePlot(label="ESG Scores Plot", x="ESG Category", y="Score", overlay_point=True)
 
+        # Textbox to display messages
+        message = gr.Textbox(label="Message", interactive=False)
+
+        # File output for CSV download
+        csv_output = gr.File(label="Download CSV")
+
         # Define the action when the "Generate ESG Plot" button is clicked
         plot_button.click(fn=fetch_esg_data, 
                           inputs=company, 
-                          outputs=plot_output)
+                          outputs=[plot_output, csv_output])
 
     return app