Update app.py

app.py

@@ -1,50 +1,60 @@
+import gradio as gr
 import yfinance as yf
 import pandas as pd
-import gradio as gr
-import os
 
-# Company dictionary mapping names to tickers
-companies = {
+# Mapping company names to their ticker symbols
+company_dict = {
     "Apple": "AAPL",
-    "Microsoft": "MSFT",
     "Google": "GOOGL",
+    "Microsoft": "MSFT",
     "Amazon": "AMZN",
     "Tesla": "TSLA"
 }
 
-def fetch_esg_data(company_name):
-    ticker = companies[company_name]
-    try:
-        stock = yf.Ticker(ticker)
-        esg_data = stock.sustainability
-        if esg_data is not None:
-            csv_file = f"{ticker}_esg_data.csv"
-            esg_data.to_csv(csv_file)
-            return f"ESG data for {company_name} saved to {csv_file}", csv_file
-        else:
-            return f"No ESG data available for {company_name}", None
-    except Exception as e:
-        return str(e), None
-
-def main():
-    company_names = list(companies.keys())
-
-    def wrapper(company_name):
-        msg, csv_file = fetch_esg_data(company_name)
-        if csv_file and os.path.exists(csv_file):
-            return msg, gr.File(csv_file)
-        return msg, None
-
-    interface = gr.Interface(
-        fn=wrapper,
-        inputs=gr.inputs.Dropdown(choices=company_names, label="Select a company"),
-        outputs=["text", "file"],
-        title="Fetch ESG Data",
-        description="Select a company to fetch its ESG data and save it to a CSV file."
-    )
-
-    interface.launch()
-
-if __name__ == "__main__":
-    main()
+# Function to fetch ESG data and save it to a CSV file
+def fetch_and_save_esg_data(company_name):
+    # Get the ticker symbol from the company name
+    ticker = company_dict[company_name]
+    
+    # Fetch ESG data from Yahoo Finance
+    stock = yf.Ticker(ticker)
+    esg_data = stock.sustainability
+
+    # If the data is available, save it to a CSV file
+    if esg_data is not None:
+        # Convert the ESG data to a pandas DataFrame
+        esg_df = pd.DataFrame(esg_data)
+        
+        # Save it to a CSV file
+        filename = f"{ticker}_esg_data.csv"
+        esg_df.to_csv(filename)
+        
+        return f"ESG data for {company_name} saved to {filename}", filename
+    else:
+        return f"No ESG data available for {company_name}", None
+
+# Gradio interface with a dropdown for company selection
+def app_interface():
+    with gr.Blocks() as app:
+        # Dropdown to select company name
+        company = gr.Dropdown(label="Select Company", choices=list(company_dict.keys()), value="Apple")
+        
+        # Button to trigger ESG data extraction
+        fetch_button = gr.Button("Fetch ESG Data")
+        
+        # Output text and download link
+        output_text = gr.Textbox(label="Output")
+        download_link = gr.File(label="Download CSV")
+
+        # Define the action when the button is clicked
+        fetch_button.click(fn=fetch_and_save_esg_data, 
+                           inputs=company, 
+                           outputs=[output_text, download_link])
+
+    return app
+
+# Launch the Gradio app
+app = app_interface()
+app.launch()
+