Update app.py

app.py

@@ -7,6 +7,12 @@ from gradio_client import Client
 import json
 from datetime import datetime
 
+# Initialize session state variables
+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'
+
 # Define the markdown variables
 Boxing_and_MMA_Commentary_and_Knowledge = """
 # Boxing and UFC Study of 1971 - 2024 The Greatest Fights History
@@ -95,7 +101,7 @@ def display_terms_with_links(terms):
     }
     for term in terms:
         links_md = ' '.join([f"[{emoji}]({url(term)})" for emoji, url in search_urls.items()])
-        st.markdown(f"**{term}** {links_md}", unsafe_allow_html=True)
+        st.markdown(f"- **{term}** {links_md}", unsafe_allow_html=True)
 
 # Function to perform AI lookup using Gradio client
 def perform_ai_lookup(query):
@@ -167,53 +173,46 @@ def file_management_sidebar():
     md_files.sort()
 
     if md_files:
-        selected_file = st.sidebar.selectbox("Select a markdown file to view/edit", md_files)
-
-        # Navigation buttons
-        file_index = md_files.index(selected_file)
-        col1, col2 = st.sidebar.columns([1,1])
-        if col1.button("Previous"):
-            if file_index > 0:
-                selected_file = md_files[file_index -1]
-                st.experimental_set_query_params(selected_file=selected_file)
-                st.experimental_rerun()
-        if col2.button("Next"):
-            if file_index < len(md_files) -1:
-                selected_file = md_files[file_index +1]
-                st.experimental_set_query_params(selected_file=selected_file)
-                st.experimental_rerun()
-
-        # Load file content
-        with open(selected_file, 'r', encoding='utf-8') as f:
-            file_content = f.read()
-
-        # Tabs for Markdown View and Code Editor
-        tab1, tab2 = st.tabs(["Markdown View", "Code Editor"])
-
-        with tab1:
-            st.markdown(f"### {selected_file}")
-            st.markdown(file_content)
-
-        with tab2:
-            edited_content = st.text_area("Edit the markdown content", file_content, height=400)
-            if st.button("Save Changes"):
-                with open(selected_file, 'w', encoding='utf-8') as f:
-                    f.write(edited_content)
-                st.success(f"Changes saved to {selected_file}")
-                st.experimental_rerun()
+        st.sidebar.markdown("### Markdown Files")
+        for idx, file in enumerate(md_files):
+            # Create a unique key for each file
+            key_base = f"file_{idx}_{file}"
+            col1, col2, col3 = st.sidebar.columns([6, 1, 1])
+            with col1:
+                st.write(file)
+            with col2:
+                if st.sidebar.button("📄", key=f"view_{key_base}"):
+                    st.session_state.selected_file = file
+                    st.session_state.view_mode = 'view'
+                    st.experimental_rerun()
+            with col3:
+                if st.sidebar.button("✏️", key=f"edit_{key_base}"):
+                    st.session_state.selected_file = file
+                    st.session_state.view_mode = 'edit'
+                    st.experimental_rerun()
+        # Option to create a new markdown file
+        if st.sidebar.button("Create New Markdown File"):
+            # Generate automatic filename
+            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+            new_filename = f"note_{timestamp}.md"
+            with open(new_filename, 'w', encoding='utf-8') as f:
+                f.write("# New Markdown File\n")
+            st.sidebar.success(f"Created new file: {new_filename}")
+            st.session_state.selected_file = new_filename
+            st.session_state.view_mode = 'edit'
+            st.experimental_rerun()
     else:
         st.sidebar.write("No markdown files found.")
-
-    # Option to create a new markdown file
-    if st.sidebar.button("Create New Markdown File"):
-        # Generate automatic filename
-        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-        new_filename = f"note_{timestamp}.md"
-        with open(new_filename, 'w', encoding='utf-8') as f:
-            f.write("# New Markdown File\n")
-        st.sidebar.success(f"Created new file: {new_filename}")
-        st.experimental_set_query_params(selected_file=new_filename)
-        st.experimental_rerun()
+        if st.sidebar.button("Create New Markdown File"):
+            # Generate automatic filename
+            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+            new_filename = f"note_{timestamp}.md"
+            with open(new_filename, 'w', encoding='utf-8') as f:
+                f.write("# New Markdown File\n")
+            st.sidebar.success(f"Created new file: {new_filename}")
+            st.session_state.selected_file = new_filename
+            st.session_state.view_mode = 'edit'
+            st.experimental_rerun()
 
 # Main application logic
 def main():
@@ -232,24 +231,70 @@ def main():
     display_terms_with_links(all_terms)
 
     # Process 'q' query parameter from the URL
-    query_params = st.experimental_get_query_params()
-    if 'q' in query_params:
-        search_query = query_params['q'][0]
-        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.")
+    try:
+        query_params = st.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.")
+                # Update file management sidebar
+                st.experimental_rerun()
+    except Exception as e:
+        st.write(f"An error occurred while processing query parameters: {e}")
+
+    # Handle 'action' and 'query' parameters
+    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 parameters
+                st.experimental_set_query_params()
+                st.experimental_rerun()
+    if 'query' in st.query_params:
+        query_list = st.query_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()
 
+    # Display the selected file
+    if 'selected_file' in st.session_state:
+        selected_file = st.session_state.selected_file
+        view_mode = st.session_state.get('view_mode', 'view')
+        if view_mode == 'view':
+            st.markdown(f"### Viewing {selected_file}")
+            with open(selected_file, 'r', encoding='utf-8') as f:
+                file_content = f.read()
+            st.markdown(file_content)
+        elif view_mode == 'edit':
+            st.markdown(f"### Editing {selected_file}")
+            with open(selected_file, 'r', encoding='utf-8') as f:
+                file_content = f.read()
+            edited_content = st.text_area("Edit the markdown content", file_content, height=400)
+            if st.button("Save Changes"):
+                with open(selected_file, 'w', encoding='utf-8') as f:
+                    f.write(edited_content)
+                st.success(f"Changes saved to {selected_file}")
+                st.experimental_rerun()
+
 if __name__ == "__main__":
     main()