Update app.py

app.py

@@ -1,52 +1,50 @@
-import gradio as gr
-my_theme = gr.Theme.from_hub("gradio/seafoam")
-
-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 = {
+    "Apple": "AAPL",
+    "Microsoft": "MSFT",
+    "Google": "GOOGL",
+    "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(ticker):
-    # 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 {ticker} saved to {filename}", filename
-    else:
-        return f"No ESG data available for {ticker}", None
-
-# List of companies (tickers) for dropdown selection
-company_list = ["AAPL", "GOOGL", "MSFT", "AMZN", "TSLA"]
-
-# Gradio interface with a dropdown for company selection
-def app_interface():
-    with gr.Blocks() as app:
-        # Dropdown to select company ticker
-        company = gr.Dropdown(label="Select Company", choices=company_list, value="AAPL")
-        
-        # 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()