Update backup4.QuerryParamsFixed..app.py

backup4.QuerryParamsFixed..app.py

@@ -2,9 +2,9 @@ import streamlit as st
 import os
 import glob
 import re
+import base64
 from urllib.parse import quote
 from gradio_client import Client
-import json
 from datetime import datetime
 
 # Initialize session state variables
@@ -81,6 +81,50 @@ Multiplayer_Custom_Hosting_Game_Servers_For_Simulated_Worlds = """
 29. Valheim
 """
 
+def sanitize_filename(text):
+    """Create a safe filename from text."""
+    # Remove or replace unsafe characters
+    safe_text = re.sub(r'[^\w\s-]', '_', text)
+    safe_text = re.sub(r'\s+', '_', safe_text)
+    return safe_text[:50]  # Limit length to 50 chars
+
+def save_ai_interaction(query, ai_result):
+    """Save AI interaction to a markdown file."""
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    safe_query = sanitize_filename(query)
+    filename = f"ai_interaction_{timestamp}_{safe_query}.md"
+    
+    # Format the content
+    content = f"""# AI Interaction - {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
+
+## Query
+{query}
+
+## AI Response
+{ai_result}
+"""
+    
+    # Save to file
+    try:
+        with open(filename, 'w', encoding='utf-8') as f:
+            f.write(content)
+        return filename
+    except Exception as e:
+        st.error(f"Error saving file: {e}")
+        return None
+
+def get_file_download_link(file_path):
+    """Generate a base64 download link for a file."""
+    try:
+        with open(file_path, 'r', encoding='utf-8') as f:
+            content = f.read()
+        b64 = base64.b64encode(content.encode()).decode()
+        filename = os.path.basename(file_path)
+        return f'<a href="data:text/markdown;base64,{b64}" download="{filename}">{filename}</a>'
+    except Exception as e:
+        st.error(f"Error creating download link: {e}")
+        return None
+
 # Function to parse markdown text and extract terms
 def extract_terms(markdown_text):
     lines = markdown_text.strip().split('\n')
@@ -108,9 +152,7 @@ def display_terms_with_links(terms):
 # Function to perform AI lookup using Gradio client
 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=query,
         llm_model_picked="mistralai/Mixtral-8x7B-Instruct-v0.1",
@@ -130,70 +172,67 @@ def perform_ai_lookup(query):
     combined_result = f"{result1}\n\n{result2}"
     return combined_result
 
-# Function to extract URLs from AI result
-def extract_urls(text):
+def display_file_content(file_path):
+    """Display file content with editing capabilities."""
     try:
-        date_pattern = re.compile(r'### (\d{2} \w{3} \d{4})')
-        abs_link_pattern = re.compile(r'\[(.*?)\]\((https://arxiv\.org/abs/\d+\.\d+)\)')
-        pdf_link_pattern = re.compile(r'\[⬇️\]\((https://arxiv\.org/pdf/\d+\.\d+)\)')
-        title_pattern = re.compile(r'### \d{2} \w{3} \d{4} \| \[(.*?)\]')
-        date_matches = date_pattern.findall(text)
-        abs_link_matches = abs_link_pattern.findall(text)
-        pdf_link_matches = pdf_link_pattern.findall(text)
-        title_matches = title_pattern.findall(text)
-
-        markdown_text = ""
-        for i in range(len(date_matches)):
-            date = date_matches[i]
-            title = title_matches[i]
-            abs_link = abs_link_matches[i][1]
-            pdf_link = pdf_link_matches[i]
-            markdown_text += f"**Date:** {date}\n\n"
-            markdown_text += f"**Title:** {title}\n\n"
-            markdown_text += f"**Abstract Link:** [{abs_link}]({abs_link})\n\n"
-            markdown_text += f"**PDF Link:** [{pdf_link}]({pdf_link})\n\n"
-            markdown_text += "---\n\n"
-        return markdown_text
-
+        with open(file_path, 'r', encoding='utf-8') as f:
+            content = f.read()
+        
+        # Display as code with line numbers
+        st.code(content, line_numbers=True)
+        
+        # Edit functionality
+        edited_content = st.text_area(
+            "Edit content",
+            content,
+            height=400,
+            key=f"edit_{os.path.basename(file_path)}"
+        )
+        
+        if st.button("Save Changes", key=f"save_{os.path.basename(file_path)}"):
+            try:
+                with open(file_path, 'w', encoding='utf-8') as f:
+                    f.write(edited_content)
+                st.success(f"Successfully saved changes to {file_path}")
+            except Exception as e:
+                st.error(f"Error saving changes: {e}")
     except Exception as e:
-        st.write(f"An error occurred in extract_urls: {e}")
-        return ''
-
-# Function to generate filename based on date and content
-def generate_filename(prefix, content):
-    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    safe_content = re.sub(r'\W+', '_', content[:50])
-    filename = f"{prefix}_{timestamp}_{safe_content}.md"
-    return filename
+        st.error(f"Error reading file: {e}")
 
-# Sidebar for file management
 def file_management_sidebar():
+    """Enhanced sidebar with file management capabilities."""
     st.sidebar.title("📁 File Management")
 
-    # 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
-
+    # Get list of .md files
+    md_files = sorted(glob.glob("*.md"))
+    st.session_state.files = md_files
+    
     if md_files:
         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])
+            col1, col2, col3, col4 = st.sidebar.columns([4, 2, 1, 1])
+            
             with col1:
                 st.write(file)
+            
             with col2:
-                if st.sidebar.button("📄", key=f"view_{key_base}"):
+                # Base64 download link
+                st.markdown(get_file_download_link(file), unsafe_allow_html=True)
+            
+            with col3:
+                # View button
+                if st.button("📄", key=f"view_{idx}"):
                     st.session_state.selected_file = file
                     st.session_state.view_mode = 'view'
-            with col3:
-                if st.sidebar.button("✏️", key=f"edit_{key_base}"):
+            
+            with col4:
+                # Edit button
+                if st.button("✏️", key=f"edit_{idx}"):
                     st.session_state.selected_file = file
                     st.session_state.view_mode = 'edit'
+
         # 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:
@@ -204,7 +243,6 @@ def file_management_sidebar():
     else:
         st.sidebar.write("No markdown files found.")
         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:
@@ -213,8 +251,6 @@ def file_management_sidebar():
             st.session_state.selected_file = new_filename
             st.session_state.view_mode = 'edit'
 
-    
-# Main application logic
 def main():
     st.title("Markdown Content with AI Lookup and File Management")
 
@@ -230,72 +266,31 @@ def main():
     all_terms = terms1 + terms2
     display_terms_with_links(all_terms)
 
-    # Process 'q' query parameter from the URL
-    try:
-        query_params = st.query_params
-        query = (query_params.get('q') or query_params.get('query') or [''])
-        if len(query) > 1: 
-            st.write(f"### Search query received: {query}")
+    # Process query parameters and AI lookup
+    query_params = st.query_params
+    query = query_params.get('q', '')
+    
+    if query:
+        st.write(f"### Search query received: {query}")
+        try:
             ai_result = perform_ai_lookup(query)
-            markdown_text = extract_urls(ai_result)
-            st.markdown("## Extracted URLs")
-            st.markdown(markdown_text)
-            filename = generate_filename(ai_result, query)
-            with open(filename, 'w', encoding='utf-8') as f:
-                f.write(markdown_text)
-            st.write(f"Generated file **{filename}** with AI results for {query}")
-            st.query_params.clear()
-            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.query_params
-    if 'action' in params:
-        action = st.query_params()['action'][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.query_params.clear()
-                
-    if 'query' in st.query_params:
-        query = st.query_params['query'][0]  # Get the query parameter
-        # Display content or image based on the query
-        #display_content_or_image(query)
-        st.markdown("To Implement:")
-        st.markdown(query)
+            
+            # Save the interaction
+            saved_file = save_ai_interaction(query, ai_result)
+            if saved_file:
+                st.success(f"Saved interaction to {saved_file}")
+                st.session_state.selected_file = saved_file
+                st.session_state.view_mode = 'view'
+        except Exception as e:
+            st.error(f"Error during AI lookup: {e}")
 
     # File management sidebar
     file_management_sidebar()
 
-    # Display the selected file
-    selected_file = st.session_state.get('selected_file')
-    if selected_file:
-        view_mode = st.session_state.get('view_mode', 'view')
-        if os.path.exists(selected_file):
-            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}")
-                    # 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.")
+    # Display selected file
+    if st.session_state.selected_file:
+        st.markdown(f"### Current File: {st.session_state.selected_file}")
+        display_file_content(st.session_state.selected_file)
 
 if __name__ == "__main__":
-    main()
+    main()