|
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 |
|
|
|
|
|
Site_Name = 'AI Knowledge Tree Builder 📈🌿 Grow Smarter with Every Click' |
|
title = "🌳✨AI Knowledge Tree Builder🛠️🤓" |
|
helpURL = 'https://huggingface.co/spaces/awacke1/AIKnowledgeTreeBuilder/' |
|
bugURL = 'https://huggingface.co/spaces/awacke1/AIKnowledgeTreeBuilder/' |
|
icons = '🌳✨🛠️🤓' |
|
st.set_page_config( |
|
page_title=title, |
|
page_icon=icons, |
|
layout="wide", |
|
initial_sidebar_state="auto", |
|
menu_items={ |
|
'Get Help': helpURL, |
|
'Report a bug': bugURL, |
|
'About': title |
|
} |
|
) |
|
|
|
|
|
|
|
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 = [] |
|
|
|
|
|
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 get_time_display(filename): |
|
"""Extract just the time portion from the filename.""" |
|
time_match = re.match(r'(\d{2}\d{2}[AP]M)', filename) |
|
if time_match: |
|
return time_match.group(1) |
|
return filename |
|
|
|
def sanitize_filename(text): |
|
"""Create a safe filename from text while preserving spaces.""" |
|
|
|
safe_text = re.sub(r'[^\w\s-]', ' ', text) |
|
|
|
safe_text = re.sub(r'\s+', ' ', safe_text) |
|
|
|
safe_text = safe_text.strip() |
|
return safe_text[:50] |
|
|
|
def generate_timestamp_filename(query): |
|
"""Generate filename with format: 1103AM 11032024 (Query).md""" |
|
|
|
central = pytz.timezone('US/Central') |
|
current_time = datetime.now(central) |
|
|
|
|
|
time_str = current_time.strftime("%I%M%p") |
|
date_str = current_time.strftime("%m%d%Y") |
|
|
|
|
|
safe_query = sanitize_filename(query) |
|
|
|
|
|
filename = f"{time_str} {date_str} ({safe_query}).md" |
|
|
|
return filename |
|
|
|
def delete_file(file_path): |
|
"""Delete a file and return success status.""" |
|
try: |
|
os.remove(file_path) |
|
return True |
|
except Exception as e: |
|
st.error(f"Error deleting file: {e}") |
|
return False |
|
|
|
def save_ai_interaction(query, ai_result, is_rerun=False): |
|
"""Save AI interaction to a markdown file with new filename format.""" |
|
filename = generate_timestamp_filename(query) |
|
|
|
|
|
if is_rerun: |
|
content = f"""# Rerun Query |
|
Original file content used for rerun: |
|
|
|
{query} |
|
|
|
# AI Response (Fun Version) |
|
{ai_result} |
|
""" |
|
else: |
|
content = f"""# Query: {query} |
|
|
|
## AI Response |
|
{ai_result} |
|
""" |
|
|
|
|
|
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}">{get_display_name(filename)}</a>' |
|
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() |
|
|
|
if st.session_state.view_mode == 'view': |
|
|
|
st.markdown(content) |
|
else: |
|
|
|
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") |
|
|
|
|
|
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("---") |
|
|
|
|
|
st.sidebar.text(get_time_display(file)) |
|
|
|
|
|
download_link = get_file_download_link(file) |
|
if download_link: |
|
st.sidebar.markdown(download_link, unsafe_allow_html=True) |
|
|
|
|
|
col1, col2, col3, col4 = st.sidebar.columns(4) |
|
|
|
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() |
|
|
|
|
|
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 |
|
|
|
|
|
ai_result = perform_ai_lookup(full_prompt) |
|
saved_file = save_ai_interaction(content, ai_result, is_rerun=True) |
|
|
|
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}") |
|
|
|
with col4: |
|
if st.button("🗑️ Delete", key=f"delete_{idx}"): |
|
if delete_file(file): |
|
st.success(f"Deleted {file}") |
|
st.rerun() |
|
else: |
|
st.error(f"Failed to delete {file}") |
|
|
|
st.sidebar.markdown("---") |
|
|
|
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("AI Knowledge Tree Builder 🧠🌱 Cultivate Your AI Mindscape!") |
|
|
|
|
|
query_params = st.query_params |
|
query = query_params.get('q', '') |
|
show_initial_content = True |
|
|
|
|
|
if query: |
|
show_initial_content = False |
|
st.write(f"### Search query received: {query}") |
|
try: |
|
ai_result = perform_ai_lookup(query) |
|
|
|
|
|
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() |
|
|
|
|
|
if st.session_state.selected_file: |
|
show_initial_content = False |
|
if os.path.exists(st.session_state.selected_file): |
|
st.markdown(f"### Current File: {st.session_state.selected_file}") |
|
display_file_content(st.session_state.selected_file) |
|
else: |
|
st.error("Selected file no longer exists.") |
|
st.session_state.selected_file = None |
|
st.rerun() |
|
|
|
|
|
if show_initial_content: |
|
|
|
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() |