perplexity update

pratham-darooka · Jun 17, 2024 · 4aa0760 · 4aa0760
1 parent e4a1491
commit 4aa0760
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/app/pages/search.py b/app/pages/search.py
@@ -68,4 +68,8 @@
                 st.title(f"_Asking AI for..._ **{st.session_state.stock_info_code.title()}**")
             st.markdown(fetch_and_parse_perplexity_output(st.session_state.stock_info_code)['llm'])
             with title:
-                st.title(f"_AI Response for..._ **{st.session_state.stock_info_code.title()}**")
+                st.title(f"_AI Response for..._ **{st.session_state.stock_info_code.title()}**")
+
+            back = st.button("Go Back!", type="primary", use_container_width=True)
+            if back:
+                st.switch_page(f'pages/{st.session_state.referrer}.py')