Update src/streamlit_app.py

src/streamlit_app.py

@@ -34,6 +34,7 @@ def parse_and_render_streamed_response(streamed):
         # Extract answer content (everything after </think>)
         answer_start = streamed.find('</think>') + len('</think>')
         answer = streamed[answer_start:].strip()
+        answer_json = json.loads(answer)
         
         # Render reasoning section
         if reasoning:
@@ -43,7 +44,7 @@ def parse_and_render_streamed_response(streamed):
         # Render answer section
         if answer:
             st.subheader("💡 Answer")
-            st.markdown(answer)
+            st.markdown(answer_json.answer)
     else:
         # No <think> tags found, render entire content as answer
         st.subheader("💡 Answer")
@@ -137,17 +138,8 @@ if user_input:
         with st.spinner("Thinking..."):
             try:
                 full_response = call_perplexity_api(user_input).json()
-                full_response_json = json.loads(full_response["choices"][0]["message"]["content"])
-                st.subheader("🤔 Reasoning")
-                if full_response_json.think:
-                    st.markdown(full_response_json.think)
-                else:
-                    st.markdown("... Reasoning not found ...")
-                st.subheader("💡 Answer")
-                if full_response_json.answer:
-                    st.markdown(full_response_json.answer)
-                else:
-                    st.markdown("... Answer not found ...")
+                full_response_content = full_response["choices"][0]["message"]["content"]
+                parse_and_render_streamed_response(full_response_content)
                 st.session_state.messages.append({"role": "assistant", "content": full_response})
             except requests.RequestException as err:
                 st.error(f"Error: {err}")