Update app.py

app.py

@@ -1,18 +1,40 @@
 import streamlit as st
-import requests
+from agents.api_agent import get_asia_tech_risk
+from agents.scraping_agent import get_earnings_news
+from agents.analysis_agent import get_aum_change, earnings_summary
+from agents.language_agent import generate_brief
 from agents.voice_agent import speech_to_text, text_to_speech
 
-st.title("🧠 Morning Market Brief - Finance Assistant")
+st.set_page_config(page_title="🧠 Finance Assistant", layout="centered")
+st.title("🧠 Morning Market Brief - Voice Assistant")
 
-option = st.radio("Input Mode", ["Voice", "Click Button"])
-if option == "Voice":
-    query = speech_to_text()
-    st.write(f"You said: {query}")
+st.markdown("Ask: _What's our risk exposure in Asia tech stocks today?_")
+
+input_mode = st.radio("Choose input mode", ["🎤 Voice", "⌨️ Button"])
+
+query = ""
+if input_mode == "🎤 Voice":
+    if st.button("Record Voice"):
+        query = speech_to_text()
+        st.write("You said:", query)
 else:
-    if st.button("Get Morning Brief"):
+    if st.button("Get Market Brief"):
         query = "What's our risk exposure in Asia tech stocks today?"
 
 if query:
-    response = requests.get("http://localhost:8000/market-brief").json()
-    st.success(response['brief'])
-    text_to_speech(response['brief'])
+    # Get agent responses
+    stocks = get_asia_tech_risk(['TSM', 'SSNLF'])
+    news = get_earnings_news()
+    earnings = earnings_summary(stocks)
+    aum_change = get_aum_change(22, 18)
+    narrative = generate_brief(query)
+
+    final_brief = (
+        f"Today, your Asia tech allocation is 22% of AUM, up from 18% yesterday.\n\n"
+        f"Earnings Summary: {earnings}.\n\n"
+        f"Top Headlines:\n" + "\n".join([f"- {h}" for h in news]) + "\n\n"
+        f"{narrative}"
+    )
+
+    st.success(final_brief)
+    text_to_speech(final_brief)