Update app.py

app.py

@@ -15,36 +15,20 @@ from urllib.parse import quote
 import streamlit as st
 import streamlit.components.v1 as components
 
-# For demonstration, from huggingface_hub
+# For demonstration, we'll import from huggingface_hub
+# (You can omit if you're not using HF or adapt your own client)
 from huggingface_hub import InferenceClient
 
-# -----------------------------------------------------
-# Ensure default MarkdownCode.md & MermaidCode.md exist
-# -----------------------------------------------------
-if not os.path.exists("MarkdownCode.md"):
-    with open("MarkdownCode.md", 'w', encoding='utf-8') as f:
-        f.write("# Default Markdown\nThis is a default Markdown file.")
-    st.rerun()
-
-if not os.path.exists("MermaidCode.md"):
-    with open("MermaidCode.md", 'w', encoding='utf-8') as f:
-        # IMPORTANT: Each click line now has a 2nd string for tooltip
-        f.write("""flowchart LR
-    %% Minimal example with correct 'click' syntax
-    %%  - "Tooltip text" between the URL and the target
-    A[Default] --> B[Example]
-    click A "/?q=Default" "Open Default" "_self"
-    click B "/?q=Example" "Open Example" "_self"
-""")
-    st.rerun()
-
 # ----------------------------
 # 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"],
@@ -55,14 +39,13 @@ roleplaying_glossary = {
     }
 }
 
+# Minimal example of a transhuman glossary
 transhuman_glossary = {
     "Neural Interfaces": ["Cortex Jack", "Mind-Machine Fusion"],
     "Cybernetics": ["Robotic Limbs", "Augmented Eyes"],
 }
 
-# ------------
-# Stub Methods
-# ------------
+# Just to demonstrate how your "search_arxiv" or "SpeechSynthesis" etc. might be placeholders
 def process_text(text):
     st.write(f"process_text called with: {text}")
 
@@ -84,6 +67,7 @@ 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
 API_URL = "https://huggingface-inference-endpoint-placeholder"
 API_KEY = "hf_XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX"
 
@@ -97,6 +81,10 @@ 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.
+    """
     search_urls = {
         "🚀🌌ArXiv": lambda k: f"/?q={quote(k)}",
         "🃏Analyst": lambda k: f"/?q={quote(k)}-{quote(PromptPrefix)}",
@@ -114,6 +102,9 @@ def display_glossary_entity(k):
 
 @st.cache_resource
 def display_glossary_grid(roleplaying_glossary):
+    """
+    Displays a glossary in columns with multiple link emojis.
+    """
     search_urls = {
         "🚀🌌ArXiv": lambda k: f"/?q={quote(k)}",
         "🃏Analyst": lambda k: f"/?q={quote(k)}-{quote(PromptPrefix)}",
@@ -164,6 +155,9 @@ def get_zip_download_link(zip_file):
     return href
 
 def get_table_download_link(file_path):
+    """
+    Creates a download link for a single file from your snippet.
+    """
     try:
         with open(file_path, 'r', encoding='utf-8') as file:
             data = file.read()
@@ -189,6 +183,9 @@ def get_file_size(file_path):
     return os.path.getsize(file_path)
 
 def compare_and_delete_files(files):
+    """
+    Compare file sizes. If duplicates exist, keep only the latest.
+    """
     if not files:
         st.warning("No files to compare.")
         return
@@ -196,6 +193,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
     for size, paths in file_sizes.items():
         if len(paths) > 1:
             latest_file = max(paths, key=os.path.getmtime)
@@ -207,16 +205,14 @@ def compare_and_delete_files(files):
 
 def FileSidebar():
     """
-    Renders the file sidebar with open/view/run/delete logic.
-    Excludes README.md from the list.
+    Renders the file sidebar with all the open/view/run/delete logic.
     """
     all_files = glob.glob("*.md")
-    # Exclude README.md
-    all_files = [f for f in all_files if f != 'README.md']
-    # Filter out short-named files if desired
+    # Example logic filtering filenames
     all_files = [file for file in all_files if len(os.path.splitext(file)[0]) >= 5]
     all_files.sort(key=lambda x: (os.path.splitext(x)[1], x), reverse=True)
 
+    # Buttons for "Delete All" and "Download"
     Files1, Files2 = st.sidebar.columns(2)
     with Files1:
         if st.button("🗑 Delete All"):
@@ -232,9 +228,11 @@ 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
@@ -260,8 +258,12 @@ 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
 
+    # Duplicate detection
     file_sizes = [get_file_size(file) for file in all_files]
     previous_size = None
     st.sidebar.title("File Operations")
@@ -281,19 +283,22 @@ def FileSidebar():
                 st.rerun()
         previous_size = size
 
+    # If we have loaded something
     if len(file_contents) > 0:
         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')
+                file_name_input = st.text_input('File Name:', file_name, key='file_name_input', on_change=None)
                 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)
@@ -306,7 +311,6 @@ def FileSidebar():
             if st.button('🔍Run'):
                 st.write("Running GPT logic placeholder...")
 
-
 # ---------------------------
 # Basic Scoring / Glossaries
 # ---------------------------
@@ -338,6 +342,9 @@ def load_score(key):
     return 0
 
 def display_buttons_with_scores(num_columns_text):
+    """
+    Show buttons that track a 'score' from your glossary data.
+    """
     game_emojis = {
         "Dungeons and Dragons": "🐉",
         "Call of Cthulhu": "🐙",
@@ -373,16 +380,19 @@ def display_buttons_with_scores(num_columns_text):
                     newscore = update_score(key.replace('?',''))
                     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'.
+    """
     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
@@ -394,6 +404,7 @@ def display_images_and_wikipedia_summaries(num_columns=4):
                 st.image(image, use_column_width=True)
                 k = image_file.split('.')[0]
                 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:
                     response = process_image(image_file, image_text_input)
@@ -403,6 +414,9 @@ def display_images_and_wikipedia_summaries(num_columns=4):
         col_index += 1
 
 def display_videos_and_links(num_columns=4):
+    """
+    Displays all .mp4 or .webm videos found in the current directory in a grid.
+    """
     video_files = [f for f in os.listdir('.') if f.endswith(('.mp4', '.webm'))]
     if not video_files:
         st.write("No MP4 or WEBM videos found in the current directory.")
@@ -417,9 +431,11 @@ 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
                     seconds_per_frame = 10
                     process_video(video_file, seconds_per_frame)
                 except ValueError:
@@ -428,15 +444,19 @@ def display_videos_and_links(num_columns=4):
 
 
 # -------------------------------------
-# Query Param Helpers
+# Query Param Helpers from your snippet
 # -------------------------------------
 def get_all_query_params(key):
-    return st.query_params.get(key, [])
+    return st.query_params().get(key, [])
 
 def clear_query_params():
-    st.query_params
+    st.query_params()
 
 def display_content_or_image(query):
+    """
+    If a query matches something in transhuman_glossary or
+    a local image, show it. Otherwise warn no match.
+    """
     for category, term_list in transhuman_glossary.items():
         for term in term_list:
             if query.lower() in term.lower():
@@ -452,9 +472,13 @@ def display_content_or_image(query):
 
 
 # ------------------------------------
-# MERMAID DIAGRAM with Tooltip Fix
+# MERMAID DIAGRAM with Clickable Links
 # ------------------------------------
 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.
+    """
     return f"""
     <html>
     <head>
@@ -466,6 +490,7 @@ 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>
@@ -482,90 +507,105 @@ 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 not model_selected:
         return url
     delimiter = "&" if "?" in url else "?"
     return f"{url}{delimiter}model=1"
 
 
+# For demonstration, we add clickable nodes & edges:
+#   click <nodeId> "<URL>" "_self"
+# If you want edges to be clickable, you can label them as well,
+# but Mermaid typically only has a 'click' property for nodes.
+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
+
+    LLM -- "Query 🔍" --> HS[Hybrid Search 🔎\\nVector+NER+Lexical]
+    click HS "/?q=Hybrid%20Search%20Vector+NER+Lexical" _self
+
+    HS -- "Reason 🤔" --> RE[Reasoning Engine 🛠️\\nNeuralNetwork+Medical]
+    click RE "/?q=Reasoning%20Engine%20NeuralNetwork+Medical" _self
+
+    RE -- "Link 📡" --> KG((Knowledge Graph 📚\\nOntology+GAR+RAG))
+    click KG "/?q=Knowledge%20Graph%20Ontology+GAR+RAG" _self
+
+    %% If you want an "edge click" to pass ?r= something, 
+    %% Mermaid doesn't have direct 'click' for edges, 
+    %% but you can define them as nodes or use linkStyle trick, etc.
+"""
+
+
+# ---------------------------
+# Streamlit Main App
+# ---------------------------
 def main():
     st.set_page_config(page_title="Mermaid + Clickable Links Demo", layout="wide")
 
-    # 1) Parse query strings using st.query_params
-    query_params = st.query_params
-    current_q = query_params.get("q", [""])[0]
+    # 1) Parse query strings on page load
+    query_params = st.experimental_get_query_params()
+    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
     st.sidebar.write("## Diagram Link Settings")
     model_selected = st.sidebar.checkbox("Append ?model=1 to each link?")
 
-    # Load from .md files
-    markdown_default = load_file("MarkdownCode.md")
-    mermaid_default = load_file("MermaidCode.md")
-
-    #  Rebuild for clickable diagram with optional &model=1
-    base_diagram = mermaid_default or ""
+    # 3) Generate a dynamic Mermaid code, appending model param if user wants
+    #    We'll do a simple string replace to incorporate the model param
+    #    For a robust approach, parse each URL carefully, then reassemble.
+    base_diagram = DEFAULT_MERMAID
     lines = base_diagram.strip().split("\n")
     new_lines = []
     for line in lines:
         if "click " in line and '"/?' in line:
-            # We expect 4 segments: e.g. click A "URL" "Tooltip" "_self"
-            parts = re.split(r'click\s+\S+\s+"([^"]+)"\s+"([^"]+)"\s+"([^"]+)"', line)
-            # parts => [prefix, URL, tooltip, target, remainder?]
-            if len(parts) == 5:
+            # e.g. click LLM "/?q=LLM%20Agent" _self
+            # let's isolate the URL part
+            parts = re.split(r'click\s+\S+\s+"([^"]+)"\s+("_self")', line)
+            if len(parts) == 4:
+                # parts[0] = 'click LLM '
+                # parts[1] = '/?q=LLM%20Agent%20Extract%20Info'
+                # parts[2] = ' _self'
+                # parts[3] = '' (trailing possibly)
                 url = parts[1]
-                tooltip = parts[2]
-                target = parts[3]
                 updated_url = append_model_param(url, model_selected)
-                # Reassemble line
-                new_line = f"{parts[0]}click {new_lines_joiner(parts[0])}\"{updated_url}\" \"{tooltip}\" \"{target}\"{parts[4]}"
-                # But we need the node name from the 'prefix' part. Let's do a simpler approach:
-                # We can do a capturing group for the node ID in a separate pattern or handle manually.
-                # For simplicity, let's just rewrite if the user used consistent 'click NodeID "URL" "Tooltip" "Target"'
-                # We'll do a second approach:
-
-                # We'll do a direct approach: parse manually with a simpler pattern
-                # Instead, let's keep it simpler for demonstration:
-                # If your usage is consistent, you can skip all this complexity and just do a standard line replacement.
-
-                new_lines.append("click fix: " + line)  # placeholder, see below
+                # Recombine
+                new_line = f"{parts[0]}\"{updated_url}\" {parts[2]}"
+                new_lines.append(new_line)
             else:
-                # Possibly different or partial usage
-                # We'll do the simpler approach: just find the quoted URL & update it
-                short_line = re.split(r'click\s+(\S+)\s+"([^"]+)"\s+"([^"]+)"\s+"([^"]+)"', line)
-                # If it doesn't match, we just leave it as is or attempt a simpler replace:
-                # We'll attempt a simpler approach with a single replace if the user always uses the second quote for the URL
-                # ...
-                updated_line = line
-                # We look for the second quoted substring if possible
-                # This can get complicated quickly, so let's keep it minimal:
-                updated_line = re.sub(r'click\s+(\S+)\s+"([^"]+)"\s+"([^"]+)"\s+"([^"]+)"',
-                                      lambda m: f'click {m.group(1)} "{append_model_param(m.group(2), model_selected)}" "{m.group(3)}" "{m.group(4)}"', 
-                                      line)
-                new_lines.append(updated_line)
+                new_lines.append(line)
         else:
             new_lines.append(line)
     mermaid_code = "\n".join(new_lines)
 
+    # 4) Render the 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)
 
-    # Show inbound ?q or ?r
+    # 5) Show what the inbound ?q / ?r was
     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
     left_col, right_col = st.columns(2)
 
     # --- Left: Markdown Editor
     with left_col:
         st.subheader("Markdown Side 📝")
         if "markdown_text" not in st.session_state:
-            st.session_state["markdown_text"] = markdown_default
-
+            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"],
@@ -573,26 +613,17 @@ def main():
         )
         st.session_state["markdown_text"] = markdown_text
 
-        colA, colB, colC, colD = st.columns(4)
+        # Button row
+        colA, colB = st.columns(2)
         with colA:
-            if st.button("🔄 Refresh"):
+            if st.button("🔄 Refresh Markdown"):
                 st.write("**Markdown** content refreshed! 🍿")
         with colB:
-            if st.button("❌ Clear"):
+            if st.button("❌ Clear Markdown"):
                 st.session_state["markdown_text"] = ""
-                st.rerun()
-        with colC:
-            if st.button("💾 File Save"):
-                with open("MarkdownCode.md", 'w', encoding='utf-8') as f:
-                    f.write(markdown_text)
-                st.success("Saved to MarkdownCode.md")
-        with colD:
-            md_filename = st.text_input("Filename for Markdown:", value="MarkdownCode.md", key="md_filename_key")
-            if st.button("💾 Save As"):
-                with open(md_filename, 'w', encoding='utf-8') as f:
-                    f.write(markdown_text)
-                st.success(f"Saved to {md_filename}")
+                st.experimental_rerun()
 
+        # Display
         st.markdown("---")
         st.markdown("**Preview:**")
         st.markdown(markdown_text)
@@ -600,41 +631,32 @@ def main():
     # --- Right: Mermaid Editor
     with right_col:
         st.subheader("Mermaid Side 🧜‍♂️")
+
         if "current_mermaid" not in st.session_state:
-            st.session_state["current_mermaid"] = mermaid_default
+            st.session_state["current_mermaid"] = mermaid_code
 
         mermaid_input = st.text_area(
             "Edit Mermaid Code:",
             value=st.session_state["current_mermaid"],
             height=300
         )
-
-        colC, colD, colE, colF = st.columns(4)
+        colC, colD = st.columns(2)
         with colC:
-            if st.button("🎨 Refresh"):
+            if st.button("🎨 Refresh Diagram"):
+                # Rebuild the diagram
                 st.session_state["current_mermaid"] = mermaid_input
                 st.write("**Mermaid** diagram refreshed! 🌈")
-                st.rerun()
+                st.experimental_rerun()
         with colD:
-            if st.button("❌ Clear "):
+            if st.button("❌ Clear Mermaid"):
                 st.session_state["current_mermaid"] = ""
-                st.rerun()
-        with colE:
-            if st.button("💾 File Save "):
-                with open("MermaidCode.md", 'w', encoding='utf-8') as f:
-                    f.write(mermaid_input)
-                st.success("Saved to MermaidCode.md")
-        with colF:
-            mermaid_filename = st.text_input("Filename for Mermaid:", value="MermaidCode.md", key="mermaid_filename_key")
-            if st.button("💾 Save As "):
-                with open(mermaid_filename, 'w', encoding='utf-8') as f:
-                    f.write(mermaid_input)
-                st.success(f"Saved to {mermaid_filename}")
+                st.experimental_rerun()
 
         st.markdown("---")
         st.markdown("**Mermaid Source:**")
         st.code(mermaid_input, language="python", line_numbers=True)
 
+    # 7) Show Sliders & image/video galleries
     st.markdown("---")
     st.header("Media Galleries")
 
@@ -644,6 +666,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)
 
+    # 8) Optional "Extended" UI
     showExtendedTextInterface = False
     if showExtendedTextInterface:
         display_glossary_grid(roleplaying_glossary)
@@ -651,9 +674,10 @@ def main():
         display_buttons_with_scores(num_columns_text)
         st.markdown("Extended text interface is on...")
 
+    # 9) Render the file sidebar
     FileSidebar()
 
-    # Random Title at bottom
+    # 10) Random Title at bottom
     titles = [
         "🧠🎭 Semantic Symphonies & Episodic Encores",
         "🌌🎼 AI Rhythms of Memory Lane",
@@ -667,13 +691,6 @@ def main():
     selected_title = random.choice(titles)
     st.markdown(f"**{selected_title}**")
 
-def new_lines_joiner(prefix_str):
-    """
-    A placeholder function if you needed to parse out the node name from prefix.
-    If not used, you can remove it. This is just a stub to illustrate
-    how you might handle the 'click NodeID "URL" "Tooltip" "Target"' pattern.
-    """
-    return prefix_str  # or parse out the node name
 
 if __name__ == "__main__":
     main()