Update app.py

app.py

@@ -417,10 +417,10 @@ def create_gradio_interface():
                     with gr.Column():
                         radar_chart_output = gr.Plot(label="Top Stocks Comparison")
                 
-                # --- DOWNLOAD BUTTON MOVED HERE (at the bottom, inside the tab) ---
+                # --- Working Download Button + File Output ---
                 download_btn = gr.Button("📥 Download Tables (.txt)", variant="secondary")
-                download_file = gr.File(visible=False)  # Hidden file output
-                
+                download_output = gr.File(label="Download")  # This will be invisible until file is ready
+            
             with gr.TabItem("Help & Information"):
                 gr.Markdown("""
                 ## How to Use This Tool
@@ -445,34 +445,35 @@ def create_gradio_interface():
                 Financial data is provided by Yahoo Finance via the yfinance package.
                 """)
         
-        # State to store results
-        scores_state = gr.State(pd.DataFrame())
-        metrics_state = gr.State(pd.DataFrame())
+        # We'll store the latest DataFrames in a simple way using a wrapper
+        last_scores = [pd.DataFrame()]
+        last_metrics = [pd.DataFrame()]
         
         def analyze_wrapper(*args):
             scores_df, metrics_df, bar_fig, radar_fig = asyncio.run(analyze_tickers(*args))
-            return scores_df, metrics_df, bar_fig, radar_fig, scores_df, metrics_df
+            last_scores[0] = scores_df
+            last_metrics[0] = metrics_df
+            return scores_df, metrics_df, bar_fig, radar_fig
         
-        def download_wrapper(scores_df, metrics_df):
-            if scores_df.empty:
+        def download_wrapper():
+            if last_scores[0].empty:
                 return None
-            return download_tables(scores_df, metrics_df)
+            return download_tables(last_scores[0], last_metrics[0])
         
         analyze_btn.click(
             analyze_wrapper,
             inputs=[tickers_input, growth_weight, value_weight, risk_weight],
-            outputs=[scores_output, metrics_output, bar_chart_output, radar_chart_output, scores_state, metrics_state]
+            outputs=[scores_output, metrics_output, bar_chart_output, radar_chart_output]
         )
         
         download_btn.click(
             download_wrapper,
-            inputs=[scores_state, metrics_state],
-            outputs=download_file
+            inputs=[],
+            outputs=download_output
         )
     
     return iface
 
-    
 # Entry point
 if __name__ == "__main__":
     logger.info("Starting Stock Analyzer app")