Update app.py

app.py

@@ -4,6 +4,41 @@ import json
 from PIL import Image
 from urllib.parse import quote  # Ensure this import is included
 
+
+
+
+    
+# Imports
+import base64
+import glob
+import json
+import math
+import openai
+import os
+import pytz
+import re
+import requests
+import streamlit as st
+import textract
+import time
+import zipfile
+import huggingface_hub
+import dotenv
+from audio_recorder_streamlit import audio_recorder
+from bs4 import BeautifulSoup
+from collections import deque
+from datetime import datetime
+from dotenv import load_dotenv
+from huggingface_hub import InferenceClient
+from io import BytesIO
+from openai import ChatCompletion
+from PyPDF2 import PdfReader
+from templates import bot_template, css, user_template
+from xml.etree import ElementTree as ET
+import streamlit.components.v1 as components  # Import Streamlit Components for HTML5
+
+
+
 # Set page configuration with a title and favicon
 st.set_page_config(
     page_title="🗺️🛠️ RolePlayAI",
@@ -28,9 +63,149 @@ with st.expander("Help / About 📚", expanded=False):
     - 🎮 **How to Play:** Start your quest with URL challenges, like `?q=Cyberpunk` or `?query=Dungeons and Dragons`.
     ''')
 
-# -----------------------------------------------------------------Art Card Sidebar:
-import base64
-import requests
+    
+# 9. Sidebar with UI controls to review and re-run prompts and continue responses
+@st.cache_resource
+def get_table_download_link(file_path):
+    with open(file_path, 'r') as file:
+        data = file.read()
+   
+    b64 = base64.b64encode(data.encode()).decode()  
+    file_name = os.path.basename(file_path)
+    ext = os.path.splitext(file_name)[1]  # get the file extension
+    if ext == '.txt':
+        mime_type = 'text/plain'
+    elif ext == '.py':
+        mime_type = 'text/plain'
+    elif ext == '.xlsx':
+        mime_type = 'text/plain'
+    elif ext == '.csv':
+        mime_type = 'text/plain'
+    elif ext == '.htm':
+        mime_type = 'text/html'
+    elif ext == '.md':
+        mime_type = 'text/markdown'
+    elif ext == '.wav':
+        mime_type = 'audio/wav'
+    else:
+        mime_type = 'application/octet-stream'  # general binary data type
+    href = f'<a href="data:{mime_type};base64,{b64}" target="_blank" download="{file_name}">{file_name}</a>'
+    return href
+
+def FileSidebar():
+    # ----------------------------------------------------- File Sidebar for Jump Gates ------------------------------------------
+
+    # Compose a file sidebar of markdown md files:
+    all_files = glob.glob("*.md")
+    all_files = [file for file in all_files if len(os.path.splitext(file)[0]) >= 10]  # exclude files with short names
+    all_files.sort(key=lambda x: (os.path.splitext(x)[1], x), reverse=True)  # sort by file type and file name in descending order
+    if st.sidebar.button("🗑 Delete All Text"):
+        for file in all_files:
+            os.remove(file)
+        st.experimental_rerun()
+    if st.sidebar.button("⬇️ Download All"):
+        zip_file = create_zip_of_files(all_files)
+        st.sidebar.markdown(get_zip_download_link(zip_file), unsafe_allow_html=True)
+    file_contents=''
+    next_action=''
+    for file in all_files:
+        col1, col2, col3, col4, col5 = st.sidebar.columns([1,6,1,1,1])  # adjust the ratio as needed
+        with col1:
+            if st.button("🌐", key="md_"+file):  # md emoji button
+                with open(file, 'r') as f:
+                    file_contents = f.read()
+                    next_action='md'
+        with col2:
+            st.markdown(get_table_download_link(file), unsafe_allow_html=True)
+        with col3:
+            if st.button("📂", key="open_"+file):  # open emoji button
+                with open(file, 'r') as f:
+                    file_contents = f.read()
+                    next_action='open'
+        with col4:
+            if st.button("🔍", key="read_"+file):  # search emoji button
+                with open(file, 'r') as f:
+                    file_contents = f.read()
+                    next_action='search'
+        with col5:
+            if st.button("🗑", key="delete_"+file):
+                os.remove(file)
+                st.experimental_rerun()
+
+                
+    if len(file_contents) > 0:
+        if next_action=='open':
+            file_content_area = st.text_area("File Contents:", file_contents, height=500)
+        if next_action=='md':
+            st.markdown(file_contents)
+
+            buttonlabel = '🔍Run with Llama and GPT.'
+            if st.button(key='RunWithLlamaandGPT', label = buttonlabel):
+                user_prompt = file_contents
+                
+                # Llama versus GPT Battle!
+                all=""
+                try:
+                    st.write('🔍Running with Llama.')
+                    response = StreamLLMChatResponse(file_contents)
+                    filename = generate_filename(user_prompt, "md")
+                    create_file(filename, file_contents, response, should_save)
+                    all=response
+                    #SpeechSynthesis(response)
+                except:
+                    st.markdown('Llama is sleeping.  Restart ETA 30 seconds.')
+                
+                # gpt
+                try:
+                    st.write('🔍Running with GPT.')
+                    response2 = chat_with_model(user_prompt, file_contents, model_choice)
+                    filename2 = generate_filename(file_contents, choice)
+                    create_file(filename2, user_prompt, response, should_save)
+                    all=all+response2
+                    #SpeechSynthesis(response2)
+                except:
+                    st.markdown('GPT is sleeping.  Restart ETA 30 seconds.')
+    
+                SpeechSynthesis(all)
+
+            
+        if next_action=='search':
+            file_content_area = st.text_area("File Contents:", file_contents, height=500)
+            st.write('🔍Running with Llama and GPT.')
+
+            user_prompt = file_contents
+            
+            # Llama versus GPT Battle!
+            all=""
+            try:
+                st.write('🔍Running with Llama.')
+                response = StreamLLMChatResponse(file_contents)
+                filename = generate_filename(user_prompt, ".md")
+                create_file(filename, file_contents, response, should_save)
+                all=response
+                #SpeechSynthesis(response)
+            except:
+                st.markdown('Llama is sleeping.  Restart ETA 30 seconds.')
+            
+            # gpt
+            try:
+                st.write('🔍Running with GPT.')
+                response2 = chat_with_model(user_prompt, file_contents, model_choice)
+                filename2 = generate_filename(file_contents, choice)
+                create_file(filename2, user_prompt, response, should_save)
+                all=all+response2
+                #SpeechSynthesis(response2)
+            except:
+                st.markdown('GPT is sleeping.  Restart ETA 30 seconds.')
+
+            SpeechSynthesis(all)
+            
+FileSidebar()
+# ----------------------------------------------------- File Sidebar for Jump Gates ------------------------------------------
+
+    
+# ---- Art Card Sidebar with Random Selection of image------------------------------
+
 def get_image_as_base64(url):
     response = requests.get(url)
     if response.status_code == 200:
@@ -38,22 +213,64 @@ def get_image_as_base64(url):
         return base64.b64encode(response.content).decode("utf-8")
     else:
         return None
-        
+
 def create_download_link(filename, base64_str):
     href = f'<a href="data:file/png;base64,{base64_str}" download="{filename}">Download Image</a>'
     return href
 
-image_url = "https://cdn-uploads.huggingface.co/production/uploads/620630b603825909dcbeba35/o1FI3G_BueAAanLBeqoo3.png"
+# List of image URLs
+image_urls = [
+    "https://cdn-uploads.huggingface.co/production/uploads/620630b603825909dcbeba35/o1FI3G_BueAAanLBeqoo3.png",
+#    "https://cdn-uploads.huggingface.co/production/uploads/620630b603825909dcbeba35/gv1xmIiXh1NGTeeV-cYF2.png",
+#    "https://cdn-uploads.huggingface.co/production/uploads/620630b603825909dcbeba35/2YsnDyc_nDNW71PPKozdN.png",
+#    "https://cdn-uploads.huggingface.co/production/uploads/620630b603825909dcbeba35/G_GkRD_IT3f14K7gWlbwi.png"
+]
 
-image_base64 = get_image_as_base64(image_url)
-if image_base64 is not None:
+# Select a random URL from the list
+selected_image_url = random.choice(image_urls)
+
+# Get the base64 encoded string of the selected image
+selected_image_base64 = get_image_as_base64(selected_image_url)
+
+if selected_image_base64 is not None:
     with st.sidebar:
         st.markdown("""### 🗺️🛠️ RolePlayAI""")
-        st.markdown(f"![image](data:image/png;base64,{image_base64})")
-        download_link = create_download_link("downloaded_image.png", image_base64)
+        # Display the image
+        st.markdown(f"![image](data:image/png;base64,{selected_image_base64})")
+        
+        # Create and display the download link
+        download_link = create_download_link("downloaded_image.png", selected_image_base64)
         st.markdown(download_link, unsafe_allow_html=True)
 else:
     st.sidebar.write("Failed to load the image.")
+
+# ---- Art Card Sidebar with random selection of image. -------------------------------
+
+    
+
+#def get_image_as_base64(url):
+#    response = requests.get(url)
+#    if response.status_code == 200:
+        # Convert the image to base64
+#        return base64.b64encode(response.content).decode("utf-8")
+#    else:
+#        return None
+        
+#def create_download_link(filename, base64_str):
+#    href = f'<a href="data:file/png;base64,{base64_str}" download="{filename}">Download Image</a>'
+#    return href
+
+#image_url = "https://cdn-uploads.huggingface.co/production/uploads/620630b603825909dcbeba35/o1FI3G_BueAAanLBeqoo3.png"
+
+#image_base64 = get_image_as_base64(image_url)
+#if image_base64 is not None:
+#    with st.sidebar:
+#        st.markdown("""### 🗺️🛠️ RolePlayAI""")
+#        st.markdown(f"![image](data:image/png;base64,{image_base64})")
+#        download_link = create_download_link("downloaded_image.png", image_base64)
+#        st.markdown(download_link, unsafe_allow_html=True)
+#else:
+#    st.sidebar.write("Failed to load the image.")
 # ------------------------------------------------------------- Art Card Sidebar
 
     
@@ -360,44 +577,6 @@ def display_content_or_image(query):
 
 
 
-
-    
-# Imports
-import base64
-import glob
-import json
-import math
-import openai
-import os
-import pytz
-import re
-import requests
-import streamlit as st
-import textract
-import time
-import zipfile
-import huggingface_hub
-import dotenv
-from audio_recorder_streamlit import audio_recorder
-from bs4 import BeautifulSoup
-from collections import deque
-from datetime import datetime
-from dotenv import load_dotenv
-from huggingface_hub import InferenceClient
-from io import BytesIO
-#from langchain.chat_models import ChatOpenAI
-#from langchain.chains import ConversationalRetrievalChain
-#from langchain.embeddings import OpenAIEmbeddings
-#from langchain.memory import ConversationBufferMemory
-#from langchain.text_splitter import CharacterTextSplitter
-#from langchain.vectorstores import FAISS
-from openai import ChatCompletion
-from PyPDF2 import PdfReader
-from templates import bot_template, css, user_template
-from xml.etree import ElementTree as ET
-import streamlit.components.v1 as components  # Import Streamlit Components for HTML5
-
-
 def add_Med_Licensing_Exam_Dataset():
     import streamlit as st
     from datasets import load_dataset
@@ -669,35 +848,6 @@ def truncate_document(document, length):
 def divide_document(document, max_length):
     return [document[i:i+max_length] for i in range(0, len(document), max_length)]
 
-# 9. Sidebar with UI controls to review and re-run prompts and continue responses
-@st.cache_resource
-def get_table_download_link(file_path):
-    with open(file_path, 'r') as file:
-        data = file.read()
-   
-    b64 = base64.b64encode(data.encode()).decode()  
-    file_name = os.path.basename(file_path)
-    ext = os.path.splitext(file_name)[1]  # get the file extension
-    if ext == '.txt':
-        mime_type = 'text/plain'
-    elif ext == '.py':
-        mime_type = 'text/plain'
-    elif ext == '.xlsx':
-        mime_type = 'text/plain'
-    elif ext == '.csv':
-        mime_type = 'text/plain'
-    elif ext == '.htm':
-        mime_type = 'text/html'
-    elif ext == '.md':
-        mime_type = 'text/markdown'
-    elif ext == '.wav':
-        mime_type = 'audio/wav'
-    else:
-        mime_type = 'application/octet-stream'  # general binary data type
-    href = f'<a href="data:{mime_type};base64,{b64}" target="_blank" download="{file_name}">{file_name}</a>'
-    return href
-
-
 def CompressXML(xml_text):
     root = ET.fromstring(xml_text)
     for elem in list(root.iter()):
@@ -1114,112 +1264,6 @@ def main():
             filename = generate_filename(user_prompt, choice)
             create_file(filename, user_prompt, response, should_save)
 
-    # Compose a file sidebar of markdown md files:
-    all_files = glob.glob("*.md")
-    all_files = [file for file in all_files if len(os.path.splitext(file)[0]) >= 10]  # exclude files with short names
-    all_files.sort(key=lambda x: (os.path.splitext(x)[1], x), reverse=True)  # sort by file type and file name in descending order
-    if st.sidebar.button("🗑 Delete All Text"):
-        for file in all_files:
-            os.remove(file)
-        st.experimental_rerun()
-    if st.sidebar.button("⬇️ Download All"):
-        zip_file = create_zip_of_files(all_files)
-        st.sidebar.markdown(get_zip_download_link(zip_file), unsafe_allow_html=True)
-    file_contents=''
-    next_action=''
-    for file in all_files:
-        col1, col2, col3, col4, col5 = st.sidebar.columns([1,6,1,1,1])  # adjust the ratio as needed
-        with col1:
-            if st.button("🌐", key="md_"+file):  # md emoji button
-                with open(file, 'r') as f:
-                    file_contents = f.read()
-                    next_action='md'
-        with col2:
-            st.markdown(get_table_download_link(file), unsafe_allow_html=True)
-        with col3:
-            if st.button("📂", key="open_"+file):  # open emoji button
-                with open(file, 'r') as f:
-                    file_contents = f.read()
-                    next_action='open'
-        with col4:
-            if st.button("🔍", key="read_"+file):  # search emoji button
-                with open(file, 'r') as f:
-                    file_contents = f.read()
-                    next_action='search'
-        with col5:
-            if st.button("🗑", key="delete_"+file):
-                os.remove(file)
-                st.experimental_rerun()
-
-                
-    if len(file_contents) > 0:
-        if next_action=='open':
-            file_content_area = st.text_area("File Contents:", file_contents, height=500)
-        if next_action=='md':
-            st.markdown(file_contents)
-
-            buttonlabel = '🔍Run with Llama and GPT.'
-            if st.button(key='RunWithLlamaandGPT', label = buttonlabel):
-                user_prompt = file_contents
-                
-                # Llama versus GPT Battle!
-                all=""
-                try:
-                    st.write('🔍Running with Llama.')
-                    response = StreamLLMChatResponse(file_contents)
-                    filename = generate_filename(user_prompt, "md")
-                    create_file(filename, file_contents, response, should_save)
-                    all=response
-                    #SpeechSynthesis(response)
-                except:
-                    st.markdown('Llama is sleeping.  Restart ETA 30 seconds.')
-                
-                # gpt
-                try:
-                    st.write('🔍Running with GPT.')
-                    response2 = chat_with_model(user_prompt, file_contents, model_choice)
-                    filename2 = generate_filename(file_contents, choice)
-                    create_file(filename2, user_prompt, response, should_save)
-                    all=all+response2
-                    #SpeechSynthesis(response2)
-                except:
-                    st.markdown('GPT is sleeping.  Restart ETA 30 seconds.')
-    
-                SpeechSynthesis(all)
-
-            
-        if next_action=='search':
-            file_content_area = st.text_area("File Contents:", file_contents, height=500)
-            st.write('🔍Running with Llama and GPT.')
-
-            user_prompt = file_contents
-            
-            # Llama versus GPT Battle!
-            all=""
-            try:
-                st.write('🔍Running with Llama.')
-                response = StreamLLMChatResponse(file_contents)
-                filename = generate_filename(user_prompt, ".md")
-                create_file(filename, file_contents, response, should_save)
-                all=response
-                #SpeechSynthesis(response)
-            except:
-                st.markdown('Llama is sleeping.  Restart ETA 30 seconds.')
-            
-            # gpt
-            try:
-                st.write('🔍Running with GPT.')
-                response2 = chat_with_model(user_prompt, file_contents, model_choice)
-                filename2 = generate_filename(file_contents, choice)
-                create_file(filename2, user_prompt, response, should_save)
-                all=all+response2
-                #SpeechSynthesis(response2)
-            except:
-                st.markdown('GPT is sleeping.  Restart ETA 30 seconds.')
-
-            SpeechSynthesis(all)
-            
-
     # Function to encode file to base64
     def get_base64_encoded_file(file_path):
         with open(file_path, "rb") as file: