Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 import os
 import glob
 import re
-from urllib.parse import quote, urlencode
+from urllib.parse import quote
 from gradio_client import Client
 import json
 from datetime import datetime
@@ -12,6 +12,8 @@ if 'selected_file' not in st.session_state:
     st.session_state.selected_file = None
 if 'view_mode' not in st.session_state:
     st.session_state.view_mode = 'view'
+if 'files' not in st.session_state:
+    st.session_state.files = []
 
 # Define the markdown variables
 Boxing_and_MMA_Commentary_and_Knowledge = """
@@ -91,13 +93,8 @@ def extract_terms(markdown_text):
 
 # Function to display terms with links
 def display_terms_with_links(terms):
-    def generate_arxiv_link(term):
-        params = {'term': term}
-        query_string = urlencode(params)
-        return f"./?{query_string}"
-
     search_urls = {
-        "🚀🌌ArXiv": lambda k: generate_arxiv_link(k),
+        "🚀🌌ArXiv": lambda k: f"/?q={quote(k)}",
         "📖": lambda k: f"https://en.wikipedia.org/wiki/{quote(k)}",
         "🔍": lambda k: f"https://www.google.com/search?q={quote(k)}",
         "▶️": lambda k: f"https://www.youtube.com/results?search_query={quote(k)}",
@@ -109,13 +106,13 @@ def display_terms_with_links(terms):
         st.markdown(f"- **{term}** {links_md}", unsafe_allow_html=True)
 
 # Function to perform AI lookup using Gradio client
-def perform_ai_lookup(term):
+def perform_ai_lookup(query):
     st.write("Performing AI Lookup...")
     # Initialize the Gradio client
     client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
     # Perform the AI lookup using the Mixtral and Mistral models
     result1 = client.predict(
-        prompt=term,
+        prompt=query,
         llm_model_picked="mistralai/Mixtral-8x7B-Instruct-v0.1",
         stream_outputs=True,
         api_name="/ask_llm"
@@ -123,7 +120,7 @@ def perform_ai_lookup(term):
     st.markdown("### Mixtral-8x7B-Instruct-v0.1 Result")
     st.markdown(result1)
     result2 = client.predict(
-        prompt=term,
+        prompt=query,
         llm_model_picked="mistralai/Mistral-7B-Instruct-v0.2",
         stream_outputs=True,
         api_name="/ask_llm"
@@ -163,38 +160,12 @@ def extract_urls(text):
         return ''
 
 # Function to generate filename based on date and content
-def generate_filename(prefix, content, ai_output):
+def generate_filename(prefix, content):
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     safe_content = re.sub(r'\W+', '_', content[:50])
-    # Get first few words of AI output
-    ai_snippet = ' '.join(ai_output.strip().split()[:5])
-    safe_ai_snippet = re.sub(r'\W+', '_', ai_snippet)
-    filename = f"{prefix}_{timestamp}_{safe_content}_{safe_ai_snippet}.md"
+    filename = f"{prefix}_{timestamp}_{safe_content}.md"
     return filename
 
-# Function to process query parameters and handle AI lookup
-def process_query_parameters():
-    term = st.query_params.get('term')
-    if term:
-        if len(term) > 1:
-            st.write(f"### Search term received: {term}")
-            # Perform AI lookup
-            ai_result = perform_ai_lookup(term)
-            # Extract URLs from AI result
-            markdown_text = extract_urls(ai_result)
-            st.markdown("## Extracted URLs")
-            st.markdown(markdown_text)
-            # Save the result as markdown file
-            filename = generate_filename("AI_Result", term, ai_result)
-            with open(filename, 'w', encoding='utf-8') as f:
-                f.write(markdown_text)
-            st.write(f"Generated file **{filename}** with AI lookup results.")
-            # Clear the query parameters
-            st.query_params.clear()
-            # Update session state
-            st.session_state.selected_file = filename
-            st.session_state.view_mode = 'view'
-
 # Sidebar for file management
 def file_management_sidebar():
     st.sidebar.title("📁 File Management")
@@ -202,6 +173,7 @@ def file_management_sidebar():
     # Get list of .md files excluding README.md
     md_files = [file for file in glob.glob("*.md") if os.path.basename(file).lower() != 'readme.md']
     md_files.sort()
+    st.session_state.files = md_files  # Update session state
 
     if md_files:
         st.sidebar.markdown("### Markdown Files")
@@ -245,9 +217,6 @@ def file_management_sidebar():
 def main():
     st.title("Markdown Content with AI Lookup and File Management")
 
-    # Process query parameters for AI lookup
-    process_query_parameters()
-
     # Display the original markdown content
     st.markdown("## Original Markdown Content")
     st.markdown(Boxing_and_MMA_Commentary_and_Knowledge)
@@ -260,20 +229,51 @@ def main():
     all_terms = terms1 + terms2
     display_terms_with_links(all_terms)
 
-    # Handle 'action' and 'query' parameters (if needed)
-    params = st.query_params
+    # Process 'q' query parameter from the URL
+    try:
+        query_params = st.experimental_get_query_params()
+        query_list = query_params.get('q') or query_params.get('query') or []
+        if query_list:
+            search_query = query_list[0]
+            if len(search_query) > 1:
+                st.write(f"### Search query received: {search_query}")
+                # Perform AI lookup
+                ai_result = perform_ai_lookup(search_query)
+                # Extract URLs from AI result
+                markdown_text = extract_urls(ai_result)
+                st.markdown("## Extracted URLs")
+                st.markdown(markdown_text)
+                # Save the result as markdown file
+                filename = generate_filename("AI_Result", search_query)
+                with open(filename, 'w', encoding='utf-8') as f:
+                    f.write(markdown_text)
+                st.write(f"Generated file **{filename}** with AI lookup results.")
+                # Clear the query parameters
+                st.experimental_set_query_params()
+                # Update session state
+                st.session_state.selected_file = filename
+                st.session_state.view_mode = 'view'
+    except Exception as e:
+        st.write(f"An error occurred while processing query parameters: {e}")
+
+    # Handle 'action' and 'query' parameters
+    params = st.experimental_get_query_params()
     if 'action' in params:
-        action = params['action']
-        if action == 'show_message':
-            st.success("Showing a message because 'action=show_message' was found in the URL.")
-        elif action == 'clear':
-            # Clear query parameters
-            st.query_params.clear()
+        action_list = 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 parameters
+                st.experimental_set_query_params()
     if 'query' in params:
-        query = params['query']
-        # Display content or image based on the query
-        st.write(f"Displaying content for query: {query}")
-        # Implement your display logic here
+        query_list = params['query']
+        if query_list:
+            query = query_list[0]
+            # Display content or image based on the query
+            st.write(f"Displaying content for query: {query}")
+            # Implement your display logic here
 
     # File management sidebar
     file_management_sidebar()
@@ -297,10 +297,12 @@ def main():
                     with open(selected_file, 'w', encoding='utf-8') as f:
                         f.write(edited_content)
                     st.success(f"Changes saved to {selected_file}")
+                    # Update the file list in session state
+                    st.session_state.files = [file for file in glob.glob("*.md") if os.path.basename(file).lower() != 'readme.md']
         else:
             st.error("Selected file does not exist.")
     else:
         st.info("No file selected.")
 
 if __name__ == "__main__":
-        main()
+    main()