read query parameters to be able to pass a deep link skipping the search main page

app.py

@@ -1,5 +1,4 @@
 import streamlit as st
-import openai
 import logging
 import sys
 import os
@@ -61,14 +60,11 @@ def scrape_the_article(url):
 
 
 def init_session() -> None:
-    st.session_state.context = None
-    st.session_state.question = None
-    st.session_state.sub_questions = None
-    st.session_state.end = False
+    if 'init' not in st.session_state:
+        st.session_state.init = True
+        st.session_state.question = None
 
-    st.session_state.messages = []
-
-    st.session_state.open_api_key = open_api_key
+        st.session_state.messages = []
 
 
 @st.cache_data
@@ -179,47 +175,72 @@ def show_question_input():
             st.button(q, on_click=on_question_button, args=[q])
 
 
+def get_query_params():
+    if 'web_url' not in st.session_state:
+        params = st.experimental_get_query_params()
+        logging.debug(f"query parameters: {params}")
+        if 'web_url' in params:
+            web_url = params['web_url'][0]
+            if len(web_url) > 0:
+                if is_valid_web_link(web_url):
+                    st.session_state.web_url = web_url
+
+
+def show_header():
+    if 'web_url' in st.session_state:
+        col1, col2 = st.columns(2)
+        col1.caption(f"discussing: {st.session_state.web_url}")
+        if 'title' in st.session_state:
+            col2.caption(f"{st.session_state.title}")
+
+
+def get_random_page():
+    return 'https://mailchi.mp/expresso/lightpeak'
+
+
 def main() -> None:
     try:
-        if 'web_url' in st.session_state:
-            col1, col2 = st.columns(2)
-            col1.caption(f"discussing: {st.session_state.web_url}")
-            if 'title' in st.session_state:
-                col2.caption(f"{st.session_state.title}")
-
-        if 'content' not in st.session_state:
-            init_session()
+        get_query_params()
+        init_session()
+        show_header()
+
+        if 'web_url' not in st.session_state:
             st.header("Doodle")
             st.image("./assets/doodle-img.jpg")
             description = """\
-                Meet 'Doodle,' your shortcut to understanding the web! Got a lengthy article you're eyeing? 
-                Just paste the link, and in an instant, Doodle delivers a crisp summary and intriguing questions for you to 
-                chew on. Want to go hands-free? Doodle's text-to-speech feature will read it to you! Why the name 'Doodle'? 
-                Just as a simple doodle can encapsulate a whole idea, we distill webpages down to their essence!
-            """
+                            Meet 'Doodle,' your shortcut to understanding the web! Got a lengthy article you're eyeing? 
+                            Just paste the link, and in an instant, Doodle delivers a crisp summary and intriguing questions for you to 
+                            chew on. Want to go hands-free? Doodle's text-to-speech feature will read it to you! Why the name 'Doodle'? 
+                            Just as a simple doodle can encapsulate a whole idea, we distill webpages down to their essence!
+                        """
             st.caption(description)
             st.divider()
 
-            content_url = st.text_input(label='Paste your link, e.g. https://expresso.today', label_visibility='collapsed',
+            content_url = st.text_input(label='Paste your link, e.g. https://expresso.today',
+                                        label_visibility='collapsed',
                                         placeholder='Paste your link, e.g. https://expresso.today')
             col1, _, _, _, col2 = st.columns(5)
             col1.button("Doodle")
             if col2.button("Random Page"):
-                content_url = 'https://mailchi.mp/expresso/lightpeak'
-
+                content_url = get_random_page()
             if len(content_url) > 0:
                 if is_valid_web_link(content_url):
-                    with st.spinner(f"reading the web page '{content_url}' ..."):
-                        st.session_state.web_url = content_url
-                        st.session_state.web_page = scrape_the_article(content_url)
-                        st.session_state.title = st.session_state.web_page['title']
-                        st.session_state.content = st.session_state.web_page['content']
+                    st.session_state.web_url = content_url
                     st.experimental_rerun()
                 else:
-                    st.warning("Whoops! That link seems to be doing the vanishing act. Could you give it another shot? Magic words: 'Valid Link, Please!' 🪄")
+                    st.warning(
+                        "Whoops! That link seems to be doing the vanishing act. Could you give it another shot? Magic words: 'Valid Link, Please!' 🪄")
+
+        elif 'content' not in st.session_state:
+            with st.spinner(f"reading the web page '{st.session_state.web_url}' ..."):
+                st.session_state.web_page = scrape_the_article(st.session_state.web_url)
+                st.session_state.title = st.session_state.web_page['title']
+                st.session_state.content = st.session_state.web_page['content']
+            st.experimental_rerun()
+
         elif 'content_summary' not in st.session_state:
             content_summary = get_content_summary(content=st.session_state.content, model_name="gpt-3.5-turbo-16k",
-                                                  api_key=st.session_state.open_api_key)
+                                                  api_key=open_api_key)
             st.session_state.content_summary = content_summary['summary']
             st.session_state.content_block_summary = [s['block_summary'] for s in content_summary['blocks']]
             st.session_state.content_block_questions = [s['block_question'] for s in content_summary['blocks']]
@@ -231,8 +252,7 @@ def main() -> None:
             st.subheader(question)
             st.divider()
             with st.spinner(f'answering the question...'):
-                answer = qa(query=question, documents_to_search=20, model_name='gpt-4',
-                            api_key=st.session_state.open_api_key)
+                answer = qa(query=question, documents_to_search=20, model_name='gpt-4', api_key=open_api_key)
                 show_audio_message(answer)
                 st.session_state.question = None
                 show_question_input()
@@ -248,7 +268,6 @@ def main() -> None:
             st.experimental_rerun()
 
 
-
 if __name__ == "__main__":
     main()