Update app.py

app.py

@@ -21,7 +21,7 @@ COMPANY_DICT = {
     "Netflix": "NFLX"
 }
 
-def fetch_esg_data(company_name: str) -> Tuple[pd.DataFrame, Optional[str], dict, dict, dict]:
+def fetch_esg_data(company_name: str) -> Tuple[Optional[str], dict, dict, dict]:
     """
     Fetch and process ESG data for the selected company.
     
@@ -30,7 +30,6 @@ def fetch_esg_data(company_name: str) -> Tuple[pd.DataFrame, Optional[str], dict
         
     Returns:
         Tuple containing:
-        - DataFrame with ESG scores
         - Path to saved CSV file
         - Three plotly figures for different visualizations
     """
@@ -44,7 +43,7 @@ def fetch_esg_data(company_name: str) -> Tuple[pd.DataFrame, Optional[str], dict
         esg_data = stock.sustainability
         
         if esg_data is None:
-            raise ValueError(f"No ESG data available for {company_name}")
+            return None, None, None, None
             
         # Process ESG data
         esg_df = pd.DataFrame(esg_data)
@@ -65,11 +64,11 @@ def fetch_esg_data(company_name: str) -> Tuple[pd.DataFrame, Optional[str], dict
         scatter_fig = create_scatter_plot(plot_df)
         bar_fig = create_bar_plot(plot_df)
         
-        return plot_df, csv_filename, line_fig, scatter_fig, bar_fig
+        return csv_filename, line_fig, scatter_fig, bar_fig
         
     except Exception as e:
         logger.error(f"Error fetching ESG data: {str(e)}")
-        return pd.DataFrame(), None, {}, {}, {}
+        return None, None, None, None
 
 def create_line_plot(df: pd.DataFrame) -> dict:
     """Create a line plot using plotly"""
@@ -151,22 +150,45 @@ def create_interface() -> gr.Blocks:
                 with gr.Column():
                     bar_plot = gr.Plot(label="ESG Score Comparison")
                     
-            # Error message display
-            error_message = gr.Markdown(visible=False)
+            status_message = gr.Textbox(
+                label="Status",
+                interactive=False,
+                visible=True
+            )
             
-            def handle_error(error):
-                return gr.Markdown.update(visible=True, value=f"⚠️ Error: {error}")
+            def process_esg_data(company_name):
+                try:
+                    csv_file, line, scatter, bar = fetch_esg_data(company_name)
+                    
+                    if csv_file is None:
+                        return (
+                            None,
+                            None,
+                            None,
+                            None,
+                            f"No ESG data available for {company_name}"
+                        )
+                    
+                    return (
+                        csv_file,
+                        line,
+                        scatter,
+                        bar,
+                        f"Successfully generated ESG analysis for {company_name}"
+                    )
+                except Exception as e:
+                    return None, None, None, None, f"Error: {str(e)}"
             
-            # Connect the button click to the fetch function
+            # Connect the button click to the process function
             plot_button.click(
-                fn=fetch_esg_data,
+                fn=process_esg_data,
                 inputs=company,
                 outputs=[
-                    error_message,
                     csv_output,
                     line_plot,
                     scatter_plot,
-                    bar_plot
+                    bar_plot,
+                    status_message
                 ],
                 api_name="generate_esg_analysis"
             )