Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| import os | |
| import glob | |
| import re | |
| import base64 | |
| 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 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') | |
| terms = [] | |
| for line in lines: | |
| line = re.sub(r'^[#*\->\d\.\s]+', '', line).strip() | |
| if line: | |
| terms.append(line) | |
| return terms | |
| # Function to display terms with links | |
| def display_terms_with_links(terms): | |
| 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) | |
| # Function to perform AI lookup using Gradio client | |
| def perform_ai_lookup(query): | |
| 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(): | |
| """Enhanced sidebar with file management capabilities.""" | |
| st.sidebar.title("📁 File Management") | |
| # 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): | |
| col1, col2, col3, col4 = st.sidebar.columns([4, 2, 1, 1]) | |
| with col1: | |
| st.write(file) | |
| with col2: | |
| # 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 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"): | |
| 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' | |
| else: | |
| st.sidebar.write("No markdown files found.") | |
| if st.sidebar.button("Create New Markdown File"): | |
| 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' | |
| def main(): | |
| st.title("Markdown Content with AI Lookup and File Management") | |
| # Display the original markdown content | |
| st.markdown("## Original Markdown Content") | |
| st.markdown(Boxing_and_MMA_Commentary_and_Knowledge) | |
| st.markdown(Multiplayer_Custom_Hosting_Game_Servers_For_Simulated_Worlds) | |
| # Parse and display terms with links | |
| st.markdown("## 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 | |
| display_terms_with_links(all_terms) | |
| # 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) | |
| # 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 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() |