Spaces:
Running
Running
| import os | |
| import sys | |
| import tempfile | |
| import time | |
| import itertools | |
| import streamlit as st | |
| import pandas as pd | |
| import requests | |
| import openai | |
| from threading import Thread | |
| # Add 'src' to Python path | |
| sys.path.append(os.path.join(os.path.dirname(__file__), 'src')) | |
| from main import run_pipeline | |
| st.set_page_config(page_title="📰 AI News Analyzer", layout="wide") | |
| st.title("🧠 AI-Powered Investing News Analyzer") | |
| # === API Key Input === | |
| st.subheader("🔐 API Keys") | |
| openai_api_key = st.text_input("OpenAI API Key", type="password").strip() | |
| tavily_api_key = st.text_input("Tavily API Key", type="password").strip() | |
| # === Topic Input === | |
| st.subheader("📈 Topics of Interest") | |
| topics_data = [] | |
| with st.form("topics_form"): | |
| topic_count = st.number_input("How many topics?", min_value=1, max_value=10, value=1, step=1) | |
| for i in range(topic_count): | |
| col1, col2 = st.columns(2) | |
| with col1: | |
| topic = st.text_input(f"Topic {i+1}", key=f"topic_{i}") | |
| with col2: | |
| days = st.number_input(f"Timespan (days)", min_value=1, max_value=30, value=7, key=f"days_{i}") | |
| topics_data.append({"topic": topic, "timespan_days": days}) | |
| submitted = st.form_submit_button("Run Analysis") | |
| # === Tabs Setup === | |
| tab_report, tab_articles, tab_insights = st.tabs(["📝 Report", "📋 Articles", "📊 Insights"]) | |
| if submitted: | |
| if not openai_api_key or not tavily_api_key or not all([td['topic'] for td in topics_data]): | |
| st.warning("Please fill in all fields.") | |
| else: | |
| # Reset old results | |
| articles_df = pd.DataFrame() | |
| insights_df = pd.DataFrame() | |
| html_paths = [] | |
| os.environ["OPENAI_API_KEY"] = openai_api_key | |
| os.environ["TAVILY_API_KEY"] = tavily_api_key | |
| df = pd.DataFrame(topics_data) | |
| with tempfile.NamedTemporaryFile(delete=False, suffix=".csv") as tmp_csv: | |
| df.to_csv(tmp_csv.name, index=False) | |
| csv_path = tmp_csv.name | |
| spinner_box = st.empty() | |
| log_box = st.empty() | |
| logs = [] | |
| rotating = True | |
| def log(msg): | |
| logs.append(msg) | |
| log_box.code("\n".join(logs)) | |
| # === Rotating UI Messages === | |
| def rotating_messages(): | |
| messages = itertools.cycle([ | |
| "🔍 Searching financial news...", | |
| "🧠 Running AI analysis...", | |
| "📊 Evaluating sentiment...", | |
| "📝 Generating report...", | |
| "💹 Finalizing insights..." | |
| ]) | |
| while rotating: | |
| spinner_box.markdown(f"⏳ {next(messages)}") | |
| time.sleep(1.5) | |
| rotator_thread = Thread(target=rotating_messages) | |
| rotator_thread.start() | |
| try: | |
| # Check API Keys | |
| try: | |
| client = openai.OpenAI(api_key=openai_api_key) | |
| client.models.list() | |
| log("✅ OpenAI API key is valid.") | |
| except Exception as e: | |
| log(f"❌ OpenAI API Key Error: {e}") | |
| rotating = False | |
| rotator_thread.join() | |
| st.stop() | |
| try: | |
| response = requests.post( | |
| "https://api.tavily.com/search", | |
| headers={"Authorization": f"Bearer {tavily_api_key}"}, | |
| json={"query": "test", "days": 1, "max_results": 1} | |
| ) | |
| if response.status_code == 200: | |
| log("✅ Tavily API key is valid.") | |
| else: | |
| log(f"❌ Tavily Key Error: {response.status_code} {response.text}") | |
| rotating = False | |
| rotator_thread.join() | |
| st.stop() | |
| except Exception as e: | |
| log(f"❌ Tavily API Key Error: {e}") | |
| rotating = False | |
| rotator_thread.join() | |
| st.stop() | |
| with st.spinner("⏳ Running analysis..."): | |
| html_paths, articles_df, insights_df = run_pipeline(csv_path, tavily_api_key, progress_callback=log) | |
| rotating = False | |
| rotator_thread.join() | |
| spinner_box.success("✅ Analysis complete!") | |
| # === Report Tab === | |
| with tab_report: | |
| st.subheader("📝 Latest Report") | |
| if html_paths: | |
| latest_report = html_paths[-1] | |
| with open(latest_report, 'r', encoding='utf-8') as f: | |
| html_content = f.read() | |
| # Download button for HTML report | |
| st.download_button( | |
| label="⬇️ Download Report (HTML)", | |
| data=html_content, | |
| file_name=os.path.basename(latest_report), | |
| mime="text/html" | |
| ) | |
| st.components.v1.html(html_content, height=600, scrolling=True) | |
| else: | |
| st.error("❌ No reports were generated.") | |
| # === Articles Tab === | |
| with tab_articles: | |
| st.subheader("📋 Articles Table") | |
| if not articles_df.empty: | |
| st.dataframe( | |
| articles_df[["Title", "URL", "Priority", "Sentiment", "Confidence", "Signal", "Date"]], | |
| use_container_width=True | |
| ) | |
| st.download_button( | |
| label="⬇️ Download Articles CSV", | |
| data=articles_df.to_csv(index=False).encode("utf-8"), | |
| file_name="articles.csv", | |
| mime="text/csv" | |
| ) | |
| else: | |
| st.info("No articles available.") | |
| # === Insights Tab === | |
| with tab_insights: | |
| st.subheader("📊 Investment Insights") | |
| if not insights_df.empty: | |
| st.dataframe(insights_df, use_container_width=True) | |
| st.download_button( | |
| label="⬇️ Download Insights CSV", | |
| data=insights_df.to_csv(index=False).encode("utf-8"), | |
| file_name="insights.csv", | |
| mime="text/csv" | |
| ) | |
| else: | |
| st.info("No insights available.") | |
| except Exception as e: | |
| rotating = False | |
| rotator_thread.join() | |
| spinner_box.error("❌ Failed.") | |
| log_box.error(f"❌ Error: {e}") | |
| # import os | |
| # import sys | |
| # import tempfile | |
| # import time | |
| # import streamlit as st | |
| # import pandas as pd | |
| # import requests | |
| # import openai | |
| # sys.path.append(os.path.join(os.path.dirname(__file__), 'src')) | |
| # from main import run_pipeline | |
| # st.set_page_config(page_title="📰 AI News Analyzer", layout="wide") | |
| # st.title("🧠 AI-Powered Investing News Analyzer") | |
| # # === API Key Input === | |
| # st.subheader("🔐 API Keys") | |
| # openai_api_key = st.text_input("OpenAI API Key", type="password").strip() | |
| # tavily_api_key = st.text_input("Tavily API Key", type="password").strip() | |
| # # === Topic Input === | |
| # st.subheader("📈 Topics of Interest") | |
| # topics_data = [] | |
| # with st.form("topics_form"): | |
| # topic_count = st.number_input("How many topics?", min_value=1, max_value=10, value=1, step=1) | |
| # for i in range(topic_count): | |
| # col1, col2 = st.columns(2) | |
| # with col1: | |
| # topic = st.text_input(f"Topic {i+1}", key=f"topic_{i}") | |
| # with col2: | |
| # days = st.number_input(f"Timespan (days)", min_value=1, max_value=30, value=7, key=f"days_{i}") | |
| # topics_data.append({"topic": topic, "timespan_days": days}) | |
| # submitted = st.form_submit_button("Run Analysis") | |
| # # === Tabs Setup === | |
| # tab_report, tab_articles, tab_insights, tab_debug = st.tabs(["📝 Report", "📋 Articles", "📊 Insights", "🛠 Debug"]) | |
| # if submitted: | |
| # if not openai_api_key or not tavily_api_key or not all([td['topic'] for td in topics_data]): | |
| # st.warning("Please fill in all fields.") | |
| # else: | |
| # articles_df = pd.DataFrame() | |
| # insights_df = pd.DataFrame() | |
| # html_paths = [] | |
| # os.environ["OPENAI_API_KEY"] = openai_api_key | |
| # os.environ["TAVILY_API_KEY"] = tavily_api_key | |
| # df = pd.DataFrame(topics_data) | |
| # with tempfile.NamedTemporaryFile(delete=False, suffix=".csv") as tmp_csv: | |
| # df.to_csv(tmp_csv.name, index=False) | |
| # csv_path = tmp_csv.name | |
| # spinner_box = st.empty() | |
| # log_box = st.empty() | |
| # logs = [] | |
| # def log(msg): | |
| # logs.append(msg) | |
| # log_box.code("\n".join(logs)) | |
| # try: | |
| # spinner_box.markdown("⏳ Checking API keys...") | |
| # # === Check OpenAI Key === | |
| # try: | |
| # client = openai.OpenAI(api_key=openai_api_key) | |
| # client.models.list() | |
| # log("✅ OpenAI API key is valid.") | |
| # except Exception as e: | |
| # log(f"❌ OpenAI API Key Error: {e}") | |
| # st.stop() | |
| # # === Check Tavily Key === | |
| # try: | |
| # response = requests.post( | |
| # "https://api.tavily.com/search", | |
| # headers={"Authorization": f"Bearer {tavily_api_key}"}, | |
| # json={"query": "test", "days": 1, "max_results": 1} | |
| # ) | |
| # if response.status_code == 200: | |
| # log("✅ Tavily API key is valid.") | |
| # else: | |
| # log(f"❌ Tavily Key Error: {response.status_code} {response.text}") | |
| # st.stop() | |
| # except Exception as e: | |
| # log(f"❌ Tavily API Key Error: {e}") | |
| # st.stop() | |
| # spinner_box.markdown("⏳ Running analysis pipeline...") | |
| # html_paths, articles_df, insights_df = run_pipeline(csv_path, tavily_api_key, progress_callback=log) | |
| # spinner_box.success("✅ Analysis complete!") | |
| # # === Report Tab === | |
| # with tab_report: | |
| # if html_paths: | |
| # for path in html_paths: | |
| # with open(path, 'r', encoding='utf-8') as f: | |
| # html_content = f.read() | |
| # st.components.v1.html(html_content, height=600, scrolling=True) | |
| # else: | |
| # st.error("❌ No reports were generated.") | |
| # # === Articles Tab === | |
| # with tab_articles: | |
| # st.subheader("📋 Articles Table") | |
| # if not articles_df.empty: | |
| # st.dataframe(articles_df[["Title", "URL", "Summary", "Priority", "Sentiment", "Confidence", "Signal", "Date"]], | |
| # use_container_width=True) | |
| # st.download_button( | |
| # label="⬇️ Download Articles CSV", | |
| # data=articles_df.to_csv(index=False).encode("utf-8"), | |
| # file_name="articles.csv", | |
| # mime="text/csv" | |
| # ) | |
| # else: | |
| # st.info("No articles available.") | |
| # # === Insights Tab === | |
| # with tab_insights: | |
| # st.subheader("📊 Top Investment Insights") | |
| # if not insights_df.empty: | |
| # st.dataframe(insights_df, use_container_width=True) | |
| # st.download_button( | |
| # label="⬇️ Download Insights CSV", | |
| # data=insights_df.to_csv(index=False).encode("utf-8"), | |
| # file_name="insights.csv", | |
| # mime="text/csv" | |
| # ) | |
| # else: | |
| # st.info("No insights available.") | |
| # # === Debug Tab === | |
| # with tab_debug: | |
| # st.subheader("🛠 Debug Log") | |
| # st.code("\n".join(logs) if logs else "No logs yet.") | |
| # except Exception as e: | |
| # spinner_box.error("❌ Failed.") | |
| # log_box.error(f"❌ Error: {e}") | |