Upload app.py

app.py

@@ -1624,6 +1624,29 @@ def ask_ai(problem: str, sympy_info: dict, history: list) -> str:
     )
 
 
+
+# ════════════════════════════════════════════════════════════════════
+# STREAMING WRAPPER — shows text word by word like ChatGPT
+# ════════════════════════════════════════════════════════════════════
+def ask_ai_streaming(problem: str, sympy_info: dict, history: list) -> str:
+    """Get AI response and stream it word by word using st.write_stream."""
+    import time
+
+    # Get full response first
+    full_response = ask_ai(problem, sympy_info, history)
+
+    # Stream it word by word
+    def word_generator():
+        words = full_response.split(" ")
+        for i, word in enumerate(words):
+            yield word + (" " if i < len(words)-1 else "")
+            time.sleep(0.015)  # 15ms per word — smooth streaming
+
+    # Use st.write_stream for streaming display
+    streamed = st.write_stream(word_generator())
+    return full_response  # return full for history
+
+
 # ════════════════════════════════════════════════════════════════════
 # SIDEBAR
 # ════════════════════════════════════════════════════════════════════
@@ -1747,12 +1770,14 @@ if problem and problem != st.session_state.last_submitted:
     with st.chat_message("user", avatar="🧑‍🎓"):
         st.markdown(problem)
 
-    # Show AI response
+    # Show AI response with streaming
     with st.chat_message("assistant", avatar="📐"):
+        # Step 1: SymPy computation (instant)
         with st.spinner("🧮 Computing..."):
             sympy_result = run_sympy(problem)
-            answer = ask_ai(problem, sympy_result, st.session_state.messages)
-        st.markdown(answer)
+
+        # Step 2: Stream AI response word by word
+        answer = ask_ai_streaming(problem, sympy_result, st.session_state.messages)
 
     # Save to history
     st.session_state.messages.append({"role": "user", "content": problem})