Update app.py

app.py

@@ -6,36 +6,40 @@ from agents.language_agent import generate_brief
 from agents.voice_agent import speech_to_text, text_to_speech
 
 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:", "What’s our risk exposure in Asia tech stocks today, and highlight any earnings surprises?")
+    user_query = st.text_input("Ask your market question", "What’s our risk exposure in Asia tech stocks today, and highlight any earnings surprises?")
 
+# LOGIC
 if st.button("Get Market Brief"):
     if user_query.strip() == "":
-        st.warning("Please provide a question.")
+        st.warning("Please provide a valid question.")
     else:
-        # Fake output for now – plug in RAG later
-        response = """
-        Today, your Asia tech allocation is 22% of AUM, up from 18% yesterday.
-        TSMC beat estimates by 4%, Samsung missed by 2%.
-        Regional sentiment is neutral with a cautionary tilt due to rising yields.
-        """
+        # get data from agents
+        asia_risk = get_asia_tech_risk()
+        earnings = get_earnings_news()
+        aum = get_aum_change()
+        earnings_summary_text = earnings_summary()
+        
+        # generate smart summary
+        final_summary = generate_brief(user_query, asia_risk, earnings, aum, earnings_summary_text)
+
+        # show
         st.subheader("📊 Market Summary")
-        st.write(response)
+        st.write(final_summary)
 
-        audio_path = text_to_speech(response)
+        # speak
+        audio_path = text_to_speech(final_summary)
         with open(audio_path, "rb") as audio_file:
             st.audio(audio_file.read(), format="audio/mp3")