Update app.py

app.py

@@ -9,6 +9,8 @@ import concurrent.futures
 import time
 from typing import Dict, List, Optional, Any, Tuple
 import logging
+import tempfile
+import os
 
 # Set up logging
 logging.basicConfig(
@@ -335,7 +337,7 @@ async def analyze_tickers(
     logger.info(f"Analysis completed in {time.time() - start_time:.2f} seconds")
     return scores_table, metrics_table, bar_chart, radar_chart
 
-# Helper: Convert DataFrame to Markdown (for LLMs)
+# Helper: Convert DataFrame to Markdown
 def dataframe_to_markdown(df: pd.DataFrame) -> str:
     if df.empty:
         return ""
@@ -345,6 +347,21 @@ def dataframe_to_markdown(df: pd.DataFrame) -> str:
     rows = ["| " + " | ".join(str(val) for val in row) + " |" for _, row in df.iterrows()]
     return "\n".join([header, separator] + rows)
 
+# NEW: Generate downloadable .txt file with both tables
+def download_tables(scores_df: pd.DataFrame, metrics_df: pd.DataFrame) -> str:
+    content = "# Stock Analysis Results\n\n"
+    content += "## Scores Table\n"
+    content += dataframe_to_markdown(scores_df) + "\n\n"
+    content += "## Financial Metrics Table\n"
+    content += dataframe_to_markdown(metrics_df) + "\n"
+    
+    # Save to temporary file
+    temp_dir = tempfile.gettempdir()
+    path = os.path.join(temp_dir, "stock_analysis_tables.txt")
+    with open(path, "w", encoding="utf-8") as f:
+        f.write(content)
+    return path
+
 # Custom CSS for better appearance
 custom_css = """
 .gradio-container {
@@ -376,7 +393,9 @@ def create_gradio_interface():
                 placeholder="AAPL, MSFT, GOOG, AMZN, TSLA",
                 lines=1
             )
-            analyze_btn = gr.Button("Analyze Stocks", variant="primary")
+            with gr.Column():
+                analyze_btn = gr.Button("Analyze Stocks", variant="primary")
+                download_btn = gr.Button("📥 Download Tables (.txt)", variant="secondary")
         
         with gr.Row():
             with gr.Column():
@@ -398,10 +417,6 @@ def create_gradio_interface():
                         bar_chart_output = gr.Plot(label="Component Scores Chart")
                     with gr.Column():
                         radar_chart_output = gr.Plot(label="Top Stocks Comparison")
-                
-                # --- NEW: Compact Markdown copy boxes (only appear after analysis) ---
-                scores_md = gr.Textbox(label="📋 Copy Scores as Markdown", value="", lines=3, max_lines=10, interactive=False)
-                metrics_md = gr.Textbox(label="📋 Copy Metrics as Markdown", value="", lines=3, max_lines=10, interactive=False)
             
             with gr.TabItem("Help & Information"):
                 gr.Markdown("""
@@ -427,16 +442,29 @@ def create_gradio_interface():
                 Financial data is provided by Yahoo Finance via the yfinance package.
                 """)
         
+        # State to store last results
+        scores_state = gr.State(pd.DataFrame())
+        metrics_state = gr.State(pd.DataFrame())
+        
         def analyze_wrapper(*args):
             scores_df, metrics_df, bar_fig, radar_fig = asyncio.run(analyze_tickers(*args))
-            scores_md_str = dataframe_to_markdown(scores_df)
-            metrics_md_str = dataframe_to_markdown(metrics_df)
-            return scores_df, metrics_df, bar_fig, radar_fig, scores_md_str, metrics_md_str
+            return scores_df, metrics_df, bar_fig, radar_fig, scores_df, metrics_df
+        
+        def download_wrapper(scores_df, metrics_df):
+            if scores_df.empty:
+                return None
+            return download_tables(scores_df, metrics_df)
         
         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_md, metrics_md]
+            outputs=[scores_output, metrics_output, bar_chart_output, radar_chart_output, scores_state, metrics_state]
+        )
+        
+        download_btn.click(
+            download_wrapper,
+            inputs=[scores_state, metrics_state],
+            outputs=gr.File()
         )
     
     return iface