Add matplotlib to requirements

app.py

@@ -1,4 +1,3 @@
-
 import os
 import sys
 import tempfile
@@ -6,10 +5,10 @@ import time
 import itertools
 import streamlit as st
 import pandas as pd
-from io import StringIO
 from threading import Thread
+from io import StringIO
 
-# Add 'src' to Python path so we can import main.py
+# Add 'src' to Python path
 sys.path.append(os.path.join(os.path.dirname(__file__), 'src'))
 from main import run_pipeline
 
@@ -51,27 +50,59 @@ if submitted:
             df.to_csv(tmp_csv.name, index=False)
             csv_path = tmp_csv.name
 
-        progress_box = st.empty()
+        # Placeholders
+        spinner_box = st.empty()
+        log_box = st.empty()
+
+        # Rotating spinner text
         rotating = True
+        logs = []
+
+        def log(msg):
+            logs.append(msg)
+            log_box.code("\n".join(logs))
 
         def rotating_messages():
             messages = itertools.cycle([
-                "🔍 Searching financial news sources...",
-                "🧠 Running language model analysis...",
-                "📊 Generating investment reports..."
+                "🔍 Searching financial news...",
+                "🧠 Running language models...",
+                "📊 Generating investment reports...",
+                "📝 Compiling markdown...",
+                "🔎 Looking for value signals..."
             ])
             while rotating:
-                progress_box.markdown(f"⏳ {next(messages)}")
-                time.sleep(1.5)
+                spinner_box.markdown(f"⏳ {next(messages)}")
+                time.sleep(1.4)
 
         rotator_thread = Thread(target=rotating_messages)
         rotator_thread.start()
 
         try:
-            output_path = run_pipeline(csv_path, tavily_api_key)
+            # Log: API checks
+            log("🔐 Checking API keys...")
+            import openai
+            openai_client = openai.OpenAI(api_key=openai_api_key)
+            openai_client.models.list()  # triggers check
+            log("✅ OpenAI API key is valid.")
+
+            import requests
+            tavily_check = requests.post(
+                "https://api.tavily.com/search",
+                headers={"Authorization": f"Bearer {tavily_api_key}"},
+                json={"query": "test", "days": 1, "max_results": 1}
+            )
+            if tavily_check.status_code == 200:
+                log("✅ Tavily API key is valid.")
+            else:
+                raise ValueError(f"Tavily key failed: {tavily_check.status_code} - {tavily_check.text}")
+
+            # Run main pipeline
+            log("🚀 Running analysis pipeline...")
+            output_path = run_pipeline(csv_path, tavily_api_key, progress_callback=log)
+
             rotating = False
             rotator_thread.join()
-            progress_box.success("✅ Analysis complete!")
+            spinner_box.success("✅ Analysis complete!")
 
             if output_path and isinstance(output_path, list):
                 for path in output_path:
@@ -93,8 +124,8 @@ if submitted:
         except Exception as e:
             rotating = False
             rotator_thread.join()
-            progress_box.error(f"❌ Error: {e}")
-
+            spinner_box.error("❌ Failed.")
+            log_box.error(f"❌ Error: {e}")
 
 ##################################################################################################
 ##################################################################################################