Update app.py

app.py

@@ -15,20 +15,17 @@ from urllib.parse import quote
 import streamlit as st
 import streamlit.components.v1 as components
 
-# For demonstration, we'll import from huggingface_hub
-# (You can omit if you're not using HF or adapt your own client)
+# (Optional) If you use huggingface_hub:
 from huggingface_hub import InferenceClient
 
+
 # ----------------------------
 # Placeholder data structures
 # ----------------------------
-
-# Example placeholders for prompt prefixes
 PromptPrefix = "AI-Search: "
 PromptPrefix2 = "AI-Refine: "
 PromptPrefix3 = "AI-JS: "
 
-# Minimal example of a roleplaying glossary
 roleplaying_glossary = {
     "Core Rulebooks": {
         "Dungeons and Dragons": ["Player's Handbook", "Dungeon Master's Guide", "Monster Manual"],
@@ -39,13 +36,15 @@ roleplaying_glossary = {
     }
 }
 
-# Minimal example of a transhuman glossary
 transhuman_glossary = {
     "Neural Interfaces": ["Cortex Jack", "Mind-Machine Fusion"],
     "Cybernetics": ["Robotic Limbs", "Augmented Eyes"],
 }
 
-# Just to demonstrate how your "search_arxiv" or "SpeechSynthesis" etc. might be placeholders
+
+# ------------------------------------------
+# Example stubs for placeholders
+# ------------------------------------------
 def process_text(text):
     st.write(f"process_text called with: {text}")
 
@@ -67,7 +66,8 @@ def process_video(video_file, seconds_per_frame):
 def search_glossary(content):
     st.write(f"search_glossary called with: {content}")
 
-# If you have HF Inference endpoint, set them here, else placeholders
+
+# If you have HF Inference endpoints, placeholders here
 API_URL = "https://huggingface-inference-endpoint-placeholder"
 API_KEY = "hf_XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX"
 
@@ -82,8 +82,7 @@ def InferenceLLM(prompt):
 @st.cache_resource
 def display_glossary_entity(k):
     """
-    Example of how you'd create multiple links for a glossary entity.
-    This was in your original snippet. We'll keep it short.
+    Creates multiple links (emojis) for a single entity.
     """
     search_urls = {
         "🚀🌌ArXiv": lambda k: f"/?q={quote(k)}",
@@ -151,8 +150,7 @@ def get_zip_download_link(zip_file):
     with open(zip_file, 'rb') as f:
         data = f.read()
     b64 = base64.b64encode(data).decode()
-    href = f'<a href="data:application/zip;base64,{b64}" download="{zip_file}">Download All</a>'
-    return href
+    return f'<a href="data:application/zip;base64,{b64}" download="{zip_file}">Download All</a>'
 
 def get_table_download_link(file_path):
     """
@@ -174,8 +172,7 @@ def get_table_download_link(file_path):
             '.wav': 'audio/wav'
         }
         mime_type = mime_map.get(ext, 'application/octet-stream')
-        href = f'<a href="data:{mime_type};base64,{b64}" target="_blank" download="{file_name}">{file_name}</a>'
-        return href
+        return f'<a href="data:{mime_type};base64,{b64}" target="_blank" download="{file_name}">{file_name}</a>'
     except:
         return ''
 
@@ -193,7 +190,7 @@ def compare_and_delete_files(files):
     for file in files:
         size = os.path.getsize(file)
         file_sizes.setdefault(size, []).append(file)
-    # Remove all but the latest file for each size
+    # Remove all but the latest file for each size group
     for size, paths in file_sizes.items():
         if len(paths) > 1:
             latest_file = max(paths, key=os.path.getmtime)
@@ -201,24 +198,24 @@ def compare_and_delete_files(files):
                 if file != latest_file:
                     os.remove(file)
                     st.success(f"Deleted {file} as a duplicate.")
-    st.rerun()
+
 
 def FileSidebar():
     """
     Renders the file sidebar with all the open/view/run/delete logic.
     """
     all_files = glob.glob("*.md")
-    # Example logic filtering filenames
-    all_files = [file for file in all_files if len(os.path.splitext(file)[0]) >= 5]
+    # Filter out short-named or undesired files, if needed:
+    all_files = [f for f in all_files if len(os.path.splitext(f)[0]) >= 5]
     all_files.sort(key=lambda x: (os.path.splitext(x)[1], x), reverse=True)
 
-    # Buttons for "Delete All" and "Download"
+    # "Delete All" and "Download" buttons
     Files1, Files2 = st.sidebar.columns(2)
     with Files1:
         if st.button("🗑 Delete All"):
             for file in all_files:
                 os.remove(file)
-            st.rerun()
+            st.experimental_rerun()  # or remove if you prefer no rerun
     with Files2:
         if st.button("⬇️ Download"):
             zip_file = create_zip_of_files(all_files)
@@ -228,11 +225,9 @@ def FileSidebar():
     file_name = ''
     next_action = ''
 
-    # Each file row
     for file in all_files:
         col1, col2, col3, col4, col5 = st.sidebar.columns([1,6,1,1,1])
         with col1:
-            # Show an emoji button to do "md"
             if st.button("🌐", key="md_"+file):
                 file_contents = load_file(file)
                 file_name = file
@@ -258,12 +253,9 @@ def FileSidebar():
         with col5:
             if st.button("🗑", key="delete_"+file):
                 os.remove(file)
-                file_name = file
-                st.rerun()
-                next_action = 'delete'
-                st.session_state['next_action'] = next_action
+                st.experimental_rerun()  # or remove if no rerun needed
 
-    # Duplicate detection
+    # Optional: show file sizes
     file_sizes = [get_file_size(file) for file in all_files]
     previous_size = None
     st.sidebar.title("File Operations")
@@ -277,28 +269,26 @@ def FileSidebar():
                         file_content = f.read()
                     st.code(file_content, language="markdown")
                 except UnicodeDecodeError:
-                    st.error("Failed to decode the file with UTF-8.")
+                    st.error("Failed to decode file with UTF-8.")
             if st.button("Delete", key=f"delete3_{file}"):
                 os.remove(file)
-                st.rerun()
+                st.experimental_rerun()
         previous_size = size
 
-    # If we have loaded something
-    if len(file_contents) > 0:
+    # If we've loaded content from a file
+    if file_contents:
         if next_action == 'open':
             open1, open2 = st.columns([0.8, 0.2])
             with open1:
-                file_name_input = st.text_input('File Name:', file_name, key='file_name_input', on_change=None)
+                file_name_input = st.text_input('File Name:', file_name, key='file_name_input')
                 file_content_area = st.text_area('File Contents:', file_contents, height=300, key='file_content_area')
 
-                # Minimal “Save” stubs
                 if st.button('💾 Save File'):
                     with open(file_name_input, 'w', encoding='utf-8') as f:
                         f.write(file_content_area)
                     st.markdown(f'Saved {file_name_input} successfully.')
 
         elif next_action == 'search':
-            # Example usage
             file_content_area = st.text_area("File Contents:", file_contents, height=500)
             user_prompt = PromptPrefix2 + file_contents
             st.markdown(user_prompt)
@@ -312,9 +302,9 @@ def FileSidebar():
                 st.write("Running GPT logic placeholder...")
 
 
-# ---------------------------
+# -------------------------------------------
 # Basic Scoring / Glossaries
-# ---------------------------
+# -------------------------------------------
 score_dir = "scores"
 os.makedirs(score_dir, exist_ok=True)
 
@@ -354,7 +344,6 @@ def display_buttons_with_scores(num_columns_text):
         "Kindred of the East": "🌅",
         "Changeling": "🍃",
     }
-
     topic_emojis = {
         "Core Rulebooks": "📚",
         "Maps & Settings": "🗺️",
@@ -382,19 +371,18 @@ def display_buttons_with_scores(num_columns_text):
                     st.markdown(f"Scored **{category} - {game} - {term}** -> {newscore}")
 
 
-# --------------------
+# --------------------------------------
 # Image & Video Grids
-# --------------------
+# --------------------------------------
 def display_images_and_wikipedia_summaries(num_columns=4):
     """
-    Display all .png images in the current directory in a grid, referencing the name as a 'keyword'.
+    Display .png images in a grid with text input prompts.
     """
     image_files = [f for f in os.listdir('.') if f.endswith('.png')]
     if not image_files:
         st.write("No PNG images found in the current directory.")
         return
 
-    # Sort by length of filename, just as an example
     image_files_sorted = sorted(image_files, key=lambda x: len(x.split('.')[0]))
     cols = st.columns(num_columns)
     col_index = 0
@@ -408,7 +396,7 @@ def display_images_and_wikipedia_summaries(num_columns=4):
                 display_glossary_entity(k)
                 # Provide a text input for user interactions
                 image_text_input = st.text_input(f"Prompt for {image_file}", key=f"image_prompt_{image_file}")
-                if len(image_text_input) > 0:
+                if image_text_input:
                     response = process_image(image_file, image_text_input)
                     st.markdown(response)
             except:
@@ -417,7 +405,7 @@ def display_images_and_wikipedia_summaries(num_columns=4):
 
 def display_videos_and_links(num_columns=4):
     """
-    Displays all .mp4 or .webm videos found in the current directory in a grid.
+    Displays all .mp4/.webm videos in the directory in a grid, with text input prompts.
     """
     video_files = [f for f in os.listdir('.') if f.endswith(('.mp4', '.webm'))]
     if not video_files:
@@ -433,11 +421,10 @@ def display_videos_and_links(num_columns=4):
             k = video_file.split('.')[0]
             st.video(video_file, format='video/mp4', start_time=0)
             display_glossary_entity(k)
-            # Provide a text input
             video_text_input = st.text_input(f"Video Prompt for {video_file}", key=f"video_prompt_{video_file}")
             if video_text_input:
                 try:
-                    # Hard-coded example
+                    # For demonstration
                     seconds_per_frame = 10
                     process_video(video_file, seconds_per_frame)
                 except ValueError:
@@ -445,19 +432,25 @@ def display_videos_and_links(num_columns=4):
         col_index += 1
 
 
-# -------------------------------------
-# Query Param Helpers from your snippet
-# -------------------------------------
-def get_all_query_params(key):
-    return st.query_params().get(key, [])
-
+# --------------------------------------
+# Query Param Helpers (No experimental)
+# --------------------------------------
 def clear_query_params():
-    st.query_params()
+    """
+    In Streamlit, there's no direct method to "clear" query params 
+    without rewriting the URL. One workaround is to do:
+
+    st.experimental_set_query_params()  # with no arguments
+
+    But if you want to avoid *all* experimental calls,
+    you can provide a link or button that leads to a new URL without params.
+    """
+    st.warning("Use a redirect or link that excludes query parameters.")
+
 
 def display_content_or_image(query):
     """
-    If a query matches something in transhuman_glossary or
-    a local image, show it. Otherwise warn no match.
+    If a query matches transhuman_glossary or a local image, display it.
     """
     for category, term_list in transhuman_glossary.items():
         for term in term_list:
@@ -478,8 +471,7 @@ def display_content_or_image(query):
 # ------------------------------------
 def generate_mermaid_html(mermaid_code: str) -> str:
     """
-    Returns HTML embedding a Mermaid diagram. We embed the code
-    in <div class="mermaid"> and center it with CSS.
+    Embeds a mermaid diagram in HTML, centered.
     """
     return f"""
     <html>
@@ -492,7 +484,6 @@ def generate_mermaid_html(mermaid_code: str) -> str:
                 margin: 20px auto;
             }}
             .mermaid {{
-                /* Let the diagram scale or otherwise style as you wish */
                 max-width: 800px;
             }}
         </style>
@@ -510,8 +501,7 @@ def generate_mermaid_html(mermaid_code: str) -> str:
 
 def append_model_param(url: str, model_selected: bool) -> str:
     """
-    If 'Model' checkbox is selected, we append '&model=1' or '?model=1' to the URL.
-    We'll handle whether the URL already has a '?' or not.
+    If 'model=1' is desired, we append it to each URL in the diagram.
     """
     if not model_selected:
         return url
@@ -519,10 +509,8 @@ def append_model_param(url: str, model_selected: bool) -> str:
     return f"{url}{delimiter}model=1"
 
 
-# Default Mermaid diagram with "click" lines
 DEFAULT_MERMAID = """
 flowchart LR
-    %% Notice we have "click LLM ..." lines:
     U((User 😎)) -- "Talk 🗣️" --> LLM[LLM Agent 🤖\\nExtract Info]
     click U "/?q=User%20😎" _self
     click LLM "/?q=LLM%20Agent%20Extract%20Info" _self
@@ -537,28 +525,62 @@ flowchart LR
     click KG "/?q=Knowledge%20Graph%20Ontology+GAR+RAG" _self
 """
 
+
 # ---------------------------
-# Streamlit Main App
+# Main Streamlit App
 # ---------------------------
 def main():
     st.set_page_config(page_title="Mermaid + Clickable Links Demo", layout="wide")
 
-    # 1) Parse query strings on page load
-    query_params = st.experimental_get_query_params()  # original usage
-    current_q = query_params.get("q", [""])[0]  # If present, first string
-    current_r = query_params.get("r", [""])[0]
-
-    # 2) Let user pick if they want to add the "model=1" param to clickable links
+    # ---------------------------------------------
+    # Query Parameter Parsing (No experimental usage)
+    # ---------------------------------------------
+    try:
+        query_params = st.query_params
+
+        # Look for 'q' or 'query'
+        query_list = (query_params.get('q') or query_params.get('query') or [''])
+        q_or_query = query_list[0] if len(query_list) > 0 else ''
+        if q_or_query.strip():
+            filesearch = PromptPrefix + q_or_query
+            st.markdown(filesearch)
+            process_text(filesearch)
+
+    except Exception as e:
+        st.markdown(" ")  # do nothing if there's an error
+
+    # If 'action' in st.query_params
+    if 'action' in st.query_params:
+        action_list = st.query_params['action']
+        if action_list:
+            action = action_list[0]
+            if action == 'show_message':
+                st.success("Showing a message because 'action=show_message' was found in the URL.")
+            elif action == 'clear':
+                clear_query_params()
+                # If you wanted a full rerun with no params, you'd do a redirect or
+                # st.experimental_set_query_params() with no arguments (but that's experimental).
+
+    # If 'query' param is present, show content or image
+    if 'query' in st.query_params:
+        query_list2 = st.query_params['query']
+        if query_list2 and len(query_list2) > 0:
+            query_val = query_list2[0]
+            display_content_or_image(query_val)
+
+    # ---------------------------------------------
+    # Let user pick if they want to add ?model=1
+    # ---------------------------------------------
     st.sidebar.write("## Diagram Link Settings")
     model_selected = st.sidebar.checkbox("Append ?model=1 to each link?")
 
-    # 3) Generate a dynamic Mermaid code, appending model param if user wants
+    # Rebuild the dynamic Mermaid code
     base_diagram = DEFAULT_MERMAID
     lines = base_diagram.strip().split("\n")
     new_lines = []
     for line in lines:
         if "click " in line and '"/?' in line:
-            # e.g. click LLM "/?q=LLM%20Agent" _self
+            # e.g. click U "/?q=User" _self
             parts = re.split(r'click\s+\S+\s+"([^"]+)"\s+("_self")', line)
             if len(parts) == 4:
                 url = parts[1]
@@ -569,22 +591,18 @@ def main():
                 new_lines.append(line)
         else:
             new_lines.append(line)
-
     mermaid_code = "\n".join(new_lines)
 
-    # 4) Render the top-centered Mermaid diagram
+    # ---------------------------------------------
+    #  Render top-centered Mermaid diagram
+    # ---------------------------------------------
     st.title("Top-Centered Mermaid Diagram with Clickable Links 🏺")
     diagram_html = generate_mermaid_html(mermaid_code)
     components.html(diagram_html, height=400, scrolling=True)
 
-    # 5) Show inbound ?q / ?r
-    if current_q:
-        st.markdown(f"**Detected Query**: `?q={current_q}`")
-        display_content_or_image(current_q)
-    if current_r:
-        st.markdown(f"**Detected Relationship**: `?r={current_r}`")
-
-    # 6) Editor Columns: Markdown & Mermaid
+    # ---------------------------------------------
+    # Two-column layout: Markdown & Mermaid Editors
+    # ---------------------------------------------
     left_col, right_col = st.columns(2)
 
     # --- Left: Markdown Editor
@@ -592,7 +610,6 @@ def main():
         st.subheader("Markdown Side 📝")
         if "markdown_text" not in st.session_state:
             st.session_state["markdown_text"] = "## Hello!\nType some *Markdown* here.\n"
-        # Text area
         markdown_text = st.text_area(
             "Edit Markdown:",
             value=st.session_state["markdown_text"],
@@ -600,7 +617,6 @@ def main():
         )
         st.session_state["markdown_text"] = markdown_text
 
-        # Button row
         colA, colB = st.columns(2)
         with colA:
             if st.button("🔄 Refresh Markdown"):
@@ -608,9 +624,8 @@ def main():
         with colB:
             if st.button("❌ Clear Markdown"):
                 st.session_state["markdown_text"] = ""
-                st.experimental_rerun()
+                st.rerun()  # non-experimental re-run if available in your Streamlit version
 
-        # Show the rendered Markdown below
         st.markdown("---")
         st.markdown("**Preview:**")
         st.markdown(markdown_text)
@@ -628,23 +643,24 @@ def main():
             height=300
         )
 
-        # A small button bar
         colC, colD = st.columns(2)
         with colC:
             if st.button("🎨 Refresh Diagram"):
                 st.session_state["current_mermaid"] = mermaid_input
                 st.write("**Mermaid** diagram refreshed! 🌈")
-                st.experimental_rerun()
+                st.rerun()
         with colD:
             if st.button("❌ Clear Mermaid"):
                 st.session_state["current_mermaid"] = ""
-                st.experimental_rerun()
+                st.rerun()
 
         st.markdown("---")
         st.markdown("**Mermaid Source:**")
         st.code(mermaid_input, language="python", line_numbers=True)
 
-    # 7) Sliders & image/video galleries
+    # ---------------------------------------------
+    # Media Galleries
+    # ---------------------------------------------
     st.markdown("---")
     st.header("Media Galleries")
 
@@ -654,7 +670,9 @@ 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)
 
-    # 8) Optional "Extended" UI
+    # ---------------------------------------------
+    # Optional Extended UI
+    # ---------------------------------------------
     showExtendedTextInterface = False
     if showExtendedTextInterface:
         display_glossary_grid(roleplaying_glossary)
@@ -662,10 +680,14 @@ def main():
         display_buttons_with_scores(num_columns_text)
         st.markdown("Extended text interface is on...")
 
-    # 9) File Sidebar
+    # ---------------------------------------------
+    # File Sidebar
+    # ---------------------------------------------
     FileSidebar()
 
-    # 10) Random Title at bottom
+    # ---------------------------------------------
+    # Random Title at the bottom
+    # ---------------------------------------------
     titles = [
         "🧠🎭 Semantic Symphonies & Episodic Encores",
         "🌌🎼 AI Rhythms of Memory Lane",