Spaces:
Running
Running
| import streamlit as st | |
| import os | |
| import glob | |
| import re | |
| import base64 | |
| import pytz | |
| from urllib.parse import quote | |
| from gradio_client import Client | |
| 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' | |
| if 'files' not in st.session_state: | |
| st.session_state.files = [] | |
| # Define the markdown variables | |
| Boxing_and_MMA_Commentary_and_Knowledge = """ | |
| # Boxing and UFC Study of 1971 - 2024 The Greatest Fights History | |
| 1. In Boxing, the most heart breaking fight in Boxing was the Boom Boom Mancini fight with Duku Kim. | |
| 2. After changes to Boxing made it more safe due to the heart break. | |
| 3. Rehydration of the brain after weight ins loss preparation for a match is life saving change. | |
| 4. Fighting went from 15 rounds to 12. | |
| # UFC By Contrast.. | |
| 1. 5 Rounds of 5 Minutes each. | |
| 2. Greatest UFC Fighters: | |
| - Jon Jones could be the greatest of all time (GOAT) since he never lost. | |
| - George St. Pierre | |
| - BJ Penn | |
| - Anderson Silva | |
| - Mighty Mouse MMA's heart at 125 pounds | |
| - Kabib retired 29 and 0 | |
| - Fedor Milliano | |
| - Alex Pereira | |
| - James Tony | |
| - Randy Couture | |
| 3. You have to Judge them in their Championship Peak | |
| 4. Chris Weidman | |
| 5. Connor McGregor | |
| 6. Leg Breaking - Shin calcification and breaking baseball bats | |
| # References: | |
| 1. Joe Rogan - Interview #2219 | |
| 2. Donald J Trump | |
| """ | |
| Multiplayer_Custom_Hosting_Game_Servers_For_Simulated_Worlds = """ | |
| # Multiplayer Simulated Worlds | |
| 1. 7 Days To Die PC | |
| 2. ARK: Survival Evolved PC | |
| 3. Arma 3 PC | |
| 4. Atlas PC | |
| 5. Conan Exiles PC | |
| 6. Craftopia PC | |
| 7. DayZ PC | |
| 8. Eco - Global Survival PC | |
| 9. Empyrion - Galactic Survival PC | |
| 10. Factorio PC | |
| 11. Farming Simulator 19 PC | |
| 12. Crossplay | |
| 13. Farming Simulator 22 | |
| 14. Last Oasis PC | |
| 15. Last Oasis Classic PC | |
| 16. Minecraft (Vanilla) PC | |
| 17. Crossplay | |
| 18. Path of Titans | |
| 19. Rust PC | |
| 20. SCP: Secret Laboratory PC | |
| 21. SCUM PC | |
| 22. Satisfactory PC | |
| 23. Satisfactory (Experimental) PC | |
| 24. Crossplay | |
| 25. Space Engineers | |
| 26. Terraria (tShock & Vanilla) PC | |
| 27. The Forest PC | |
| 28. Crossplay | |
| 29. Valheim | |
| """ | |
| def get_display_name(filename): | |
| """Extract text from parentheses or return filename as is.""" | |
| match = re.search(r'\((.*?)\)', filename) | |
| if match: | |
| return match.group(1) | |
| return filename | |
| def sanitize_filename(text): | |
| """Create a safe filename from text while preserving spaces.""" | |
| # First replace unsafe characters with spaces | |
| safe_text = re.sub(r'[^\w\s-]', ' ', text) | |
| # Remove any multiple spaces | |
| safe_text = re.sub(r'\s+', ' ', safe_text) | |
| # Trim leading/trailing spaces | |
| safe_text = safe_text.strip() | |
| return safe_text[:50] # Limit length to 50 chars | |
| def generate_timestamp_filename(query): | |
| """Generate filename with format: 1103AM 11032024 (Query).md""" | |
| # Get current time in Central timezone | |
| central = pytz.timezone('US/Central') | |
| current_time = datetime.now(central) | |
| # Format the timestamp parts | |
| time_str = current_time.strftime("%I%M%p") # 1103AM format | |
| date_str = current_time.strftime("%m%d%Y") # 11032024 format | |
| # Clean up the query for filename - now preserving spaces | |
| safe_query = sanitize_filename(query) | |
| # Construct filename: "1103AM 11032024 (Input with spaces).md" | |
| filename = f"{time_str} {date_str} ({safe_query}).md" | |
| return filename | |
| def save_ai_interaction(query, ai_result): | |
| """Save AI interaction to a markdown file with new filename format.""" | |
| filename = generate_timestamp_filename(query) | |
| # Format the content | |
| content = f"""# 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() | |
| return f"data:text/markdown;base64,{b64}" | |
| except Exception as e: | |
| st.error(f"Error creating download link: {e}") | |
| return None | |
| def extract_terms(markdown_text): | |
| """Parse markdown text and extract terms.""" | |
| lines = markdown_text.strip().split('\n') | |
| terms = [] | |
| for line in lines: | |
| line = re.sub(r'^[#*\->\d\.\s]+', '', line).strip() | |
| if line: | |
| terms.append(line) | |
| return terms | |
| def display_terms_with_links(terms): | |
| """Display terms with various search links.""" | |
| search_urls = { | |
| "🚀🌌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)}", | |
| "🔎": lambda k: f"https://www.bing.com/search?q={quote(k)}", | |
| "🐦": lambda k: f"https://twitter.com/search?q={quote(k)}", | |
| } | |
| 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) | |
| def perform_ai_lookup(query): | |
| """Perform AI lookup using Gradio client.""" | |
| st.write("Performing AI Lookup...") | |
| client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern") | |
| result1 = client.predict( | |
| prompt=query, | |
| llm_model_picked="mistralai/Mixtral-8x7B-Instruct-v0.1", | |
| stream_outputs=True, | |
| api_name="/ask_llm" | |
| ) | |
| st.markdown("### Mixtral-8x7B-Instruct-v0.1 Result") | |
| st.markdown(result1) | |
| result2 = client.predict( | |
| prompt=query, | |
| llm_model_picked="mistralai/Mistral-7B-Instruct-v0.2", | |
| stream_outputs=True, | |
| api_name="/ask_llm" | |
| ) | |
| st.markdown("### Mistral-7B-Instruct-v0.2 Result") | |
| st.markdown(result2) | |
| combined_result = f"{result1}\n\n{result2}" | |
| return combined_result | |
| def display_file_content(file_path): | |
| """Display file content with editing capabilities.""" | |
| try: | |
| 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.error(f"Error reading file: {e}") | |
| def file_management_sidebar(): | |
| """Redesigned sidebar with improved layout and additional functionality.""" | |
| st.sidebar.title("📁 File Management") | |
| # Get list of .md files excluding README.md | |
| md_files = [file for file in glob.glob("*.md") if file.lower() != 'readme.md'] | |
| md_files.sort() | |
| st.session_state.files = md_files | |
| if md_files: | |
| st.sidebar.markdown("### Saved Files") | |
| for idx, file in enumerate(md_files): | |
| st.sidebar.markdown("---") # Separator between files | |
| # Display filename | |
| st.sidebar.text(file) | |
| # Display download link with simplified text | |
| display_name = get_display_name(file) | |
| download_link = get_file_download_link(file) | |
| if download_link: | |
| st.sidebar.markdown(f"📥 [{display_name}]({download_link})", unsafe_allow_html=True) | |
| # Action buttons in a row | |
| col1, col2, col3 = st.sidebar.columns(3) | |
| with col1: | |
| if st.button("📄 View", key=f"view_{idx}"): | |
| st.session_state.selected_file = file | |
| st.session_state.view_mode = 'view' | |
| with col2: | |
| if st.button("✏️ Edit", key=f"edit_{idx}"): | |
| st.session_state.selected_file = file | |
| st.session_state.view_mode = 'edit' | |
| with col3: | |
| if st.button("🔄 Rerun", key=f"rerun_{idx}"): | |
| try: | |
| with open(file, 'r', encoding='utf-8') as f: | |
| content = f.read() | |
| # Prepare the prompt with the prefix | |
| rerun_prefix = """For the markdown below reduce the text to a humorous fun outline with emojis and markdown outline levels in outline that convey all the facts and adds wise quotes and funny statements to engage the reader: | |
| """ | |
| full_prompt = rerun_prefix + content | |
| # Perform AI lookup and save results | |
| ai_result = perform_ai_lookup(full_prompt) | |
| saved_file = save_ai_interaction(f"Rerun of {file}", ai_result) | |
| if saved_file: | |
| st.success(f"Created fun version in {saved_file}") | |
| st.session_state.selected_file = saved_file | |
| st.session_state.view_mode = 'view' | |
| except Exception as e: | |
| st.error(f"Error during rerun: {e}") | |
| st.sidebar.markdown("---") | |
| # Option to create a new markdown file | |
| if st.sidebar.button("📝 Create New Note"): | |
| filename = generate_timestamp_filename("New Note") | |
| with open(filename, 'w', encoding='utf-8') as f: | |
| f.write("# New Markdown File\n") | |
| st.sidebar.success(f"Created: {filename}") | |
| st.session_state.selected_file = filename | |
| st.session_state.view_mode = 'edit' | |
| else: | |
| st.sidebar.write("No markdown files found.") | |
| if st.sidebar.button("📝 Create First Note"): | |
| filename = generate_timestamp_filename("New Note") | |
| with open(filename, 'w', encoding='utf-8') as f: | |
| f.write("# New Markdown File\n") | |
| st.sidebar.success(f"Created: {filename}") | |
| st.session_state.selected_file = filename | |
| st.session_state.view_mode = 'edit' | |
| def main(): | |
| st.title("Markdown Content with AI Lookup and File Management") | |
| # Process query parameters and AI lookup first | |
| query_params = st.query_params | |
| query = query_params.get('q', '') | |
| show_initial_content = True # Flag to control initial content display | |
| # First priority: Handle active query | |
| if query: | |
| show_initial_content = False # Hide initial content when showing query results | |
| st.write(f"### Search query received: {query}") | |
| try: | |
| ai_result = perform_ai_lookup(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() | |
| # Second priority: Display selected file content if any | |
| if st.session_state.selected_file: | |
| show_initial_content = False # Hide initial content when showing file content | |
| st.markdown(f"### Current File: {st.session_state.selected_file}") | |
| display_file_content(st.session_state.selected_file) | |
| # Show initial content: Either when first landing or when no interactive elements are active | |
| if show_initial_content: | |
| # First show the clickable terms with links | |
| terms1 = extract_terms(Boxing_and_MMA_Commentary_and_Knowledge) | |
| terms2 = extract_terms(Multiplayer_Custom_Hosting_Game_Servers_For_Simulated_Worlds) | |
| all_terms = terms1 + terms2 | |
| col1, col2 = st.columns(2) | |
| with col1: | |
| st.markdown("### Boxing & MMA") | |
| st.markdown(Boxing_and_MMA_Commentary_and_Knowledge) | |
| st.markdown("#### Related Links") | |
| display_terms_with_links(terms1) | |
| with col2: | |
| st.markdown("### Multiplayer Games") | |
| st.markdown(Multiplayer_Custom_Hosting_Game_Servers_For_Simulated_Worlds) | |
| st.markdown("#### Related Links") | |
| display_terms_with_links(terms2) | |
| if __name__ == "__main__": | |
| main() |