before prev commit - working

src/streamlit_app.py

@@ -98,7 +98,7 @@ if st.session_state.processing:
             previous_messages = [msg for msg in full_history if msg["role"] == "assistant" or 
                                (msg["role"] == "user" and msg["content"] != last_user_message)]
             
-            reply, explanation, evidence = orchestrator_chat(
+            reply, explanation, follow_up_questions, evidence = orchestrator_chat(
                 previous_messages,
                 last_user_message,
                 use_rag=st.session_state.use_rag,
@@ -109,11 +109,17 @@ if st.session_state.processing:
             cleaned_reply = remove_reasoning_and_sources(reply)
             cleaned_explanation = clean_explanation(explanation) if explanation else ""
             
+            # Format follow-up questions as a string
+            formatted_follow_up = ""
+            if follow_up_questions:
+                formatted_follow_up = "\n".join([f"{i+1}. {q}" for i, q in enumerate(follow_up_questions)])
+            
             # Add assistant response to history
             assistant_message = {
                 "role": "assistant", 
                 "content": cleaned_reply,
                 "explanation": cleaned_explanation,
+                "follow_up_questions": formatted_follow_up,
                 "evidence": evidence if evidence else []
             }
             
@@ -138,5 +144,4 @@ if __name__ == "__main__":
     import sys
     current_dir = os.path.dirname(os.path.abspath(__file__))
     if current_dir not in sys.path:
-        sys.path.insert(0, current_dir)
-        
+        sys.path.insert(0, current_dir)