Update app.py

app.py

@@ -5,45 +5,35 @@ 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.set_page_config(page_title="🧠 Finance Assistant", layout="centered")
 st.title("🧠 Morning Market Brief - Voice Assistant")
-st.markdown("### Choose input mode")
-
-input_mode = st.radio("Select input type:", ("🎤 Voice", "⌨️ Text"))
-user_query = ""
-
-# INPUT
-if input_mode == "🎤 Voice":
-    uploaded_audio = st.file_uploader("Upload your voice (WAV format)", type=["wav"])
-    if uploaded_audio is not None:
-        st.audio(uploaded_audio, format='audio/wav')
-        user_query = speech_to_text(uploaded_audio)
-        st.success(f"Transcribed Query: **{user_query}**")
-elif input_mode == "⌨️ Text":
-    user_query = st.text_input("Ask your market question", "What’s our risk exposure in Asia tech stocks today, and highlight any earnings surprises?")
-
 
+st.markdown("Ask: _What’s our risk exposure in Asia tech stocks today, and highlight any earnings surprises?_")
 
+input_mode = st.radio("Choose input mode", ["🎤 Voice", "⌨️ Button"])
 
-
-# LOGIC
-if st.button("Get Market Brief"):
-    if user_query.strip() == "":
-        st.warning("Please provide a valid question.")
-    else:
-        # get data from agents
-        asia_risk = get_asia_tech_risk(['TSM', 'SSNLF'])
-        earnings = get_earnings_news()
-        aum = get_aum_change(22,18)
-        earnings_summary_text = earnings_summary(asia_risk)
-        
-        # generate smart summary
-        final_summary = generate_brief(user_query, asia_risk, earnings, aum, earnings_summary_text)
-
-        # show
-        st.subheader("📊 Market Summary")
-        st.write(final_summary)
-
-        # speak
-        audio_path = text_to_speech(final_summary)
-        with open(audio_path, "rb") as audio_file:
-            st.audio(audio_file.read(), format="audio/mp3") 
+query = ""
+if input_mode == "🎤 Voice":
+    if st.button("Record Voice"):
+        query = speech_to_text()
+        st.write("You said:", query)
+else:
+    if st.button("Get Market Brief"):
+        query = "What's our risk exposure in Asia tech stocks today?"
+
+if query:
+    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)