Update dashboard.py

dashboard.py

@@ -83,9 +83,13 @@ with gr.Blocks(title="Quantitative AI Stock Analyzer") as demo:
             opt_btn = gr.Button("Run Weight Optimization Engine (scipy.optimize)", variant="stop")
             opt_status = gr.Textbox(label="Optimization Status")
 
+            # Add this new File component for downloading
+            opt_download = gr.File(label="Download Optimized Weights (.json)")
+
     run_btn.click(run_analysis, outputs=[table, sector_panel, nifty_trend])
     chart_btn.click(load_chart, inputs=chart_ticker, outputs=chart)
-    opt_btn.click(trigger_optimization, outputs=opt_status)
+    # Update this to handle both the message and the file download
+    opt_btn.click(trigger_optimization, outputs=[opt_status, opt_download])
 
 demo.queue(default_concurrency_limit=1)