Update app.py

app.py

@@ -421,7 +421,6 @@ def inject_base_url(url: str) -> str:
         return url
     return f"{BASE_URL}{url}"
 
-# Our default diagram, containing the "click" lines with /?q=...
 DEFAULT_MERMAID = """
 flowchart LR
     U((User 😎)) -- "Talk 🗣️" --> LLM[LLM Agent 🤖\\nExtract Info]
@@ -453,9 +452,8 @@ def main():
         st.markdown(search_payload)
         process_text(search_payload)
 
-    # If an 'action' param is present
-    if 'action' in query_params:
-        action_list = query_params['action']
+    if 'action' in st.query_params:
+        action_list = st.query_params['action']
         if action_list:
             action = action_list[0]
             if action == 'show_message':
@@ -464,8 +462,8 @@ def main():
                 clear_query_params()
 
     # If a 'query' param is present, show content or image
-    if 'query' in query_params:
-        query_val = query_params['query'][0]
+    if 'query' in st.query_params:
+        query_val = st.query_params['query'][0]
         display_content_or_image(query_val)
 
     # ---------------------------------------------
@@ -480,29 +478,23 @@ def main():
     base_diagram = DEFAULT_MERMAID
     lines = base_diagram.strip().split("\n")
     new_lines = []
-
     for line in lines:
-        # We look for lines like: click SOMENODE "/?q=Something" _self
         if "click " in line and '"/?' in line:
-            # Try to extract the URL part
-            parts = re.split(r'click\s+\S+\s+"([^"]+)"\s+("_self")', line)
-            if len(parts) == 4:
-                # Example:
-                # parts[0] -> 'click LLM '
-                # parts[1] -> '/?q=LLM%20Agent%20Extract%20Info'
-                # parts[2] -> ' _self'
-                # parts[3] -> '' or trailing
-                old_url = parts[1]
-                # 1) Prepend base if needed
+            parts = re.split(r'click\s+\S+\s+"([^"]+)"\s+(\S+)', line.strip())
+            # e.g. line: click U "/?q=User%20😎" _self
+            # parts might be:
+            #   [ 'click U ', '/?q=User%20😎', '_self', '' ]
+            if len(parts) >= 3:
+                prefix = parts[0]   # e.g. "click U "
+                old_url = parts[1] # e.g. /?q=User%20😎
+                target = parts[2]  # e.g. _self or _blank
+
                 new_url = inject_base_url(old_url)
-                # 2) Possibly add &model=1
                 new_url = append_model_param(new_url, model_selected)
 
-                # Recombine
-                new_line = f"{parts[0]}\"{new_url}\" {parts[2]}"
+                new_line = f'{prefix}"{new_url}" {target}'
                 new_lines.append(new_line)
             else:
-                # If we can't parse it, keep it as is
                 new_lines.append(line)
         else:
             new_lines.append(line)
@@ -550,7 +542,6 @@ def main():
     with right_col:
         st.subheader("Mermaid Side 🧜‍♂️")
 
-        # We store the final code in session state, so user can edit
         if "current_mermaid" not in st.session_state:
             st.session_state["current_mermaid"] = mermaid_code
 
@@ -587,7 +578,7 @@ def main():
     num_columns_video = st.slider("Choose Number of Video Columns", 1, 15, 5, key="num_columns_video")
     display_videos_and_links(num_columns_video)
 
-    # (Optionally) Extended text interface
+    # If you want extended UI for text or glossary
     showExtendedTextInterface = False
     if showExtendedTextInterface:
         # For example: