Update app.py

app.py

@@ -18,7 +18,6 @@ from xml.etree import ElementTree as ET
 from bs4 import BeautifulSoup
 from collections import deque
 from audio_recorder_streamlit import audio_recorder
-
 from dotenv import load_dotenv
 from PyPDF2 import PdfReader
 from langchain.text_splitter import CharacterTextSplitter
@@ -31,6 +30,7 @@ from templates import css, bot_template, user_template
 
 
 
+
 def generate_filename(prompt, file_type):
     central = pytz.timezone('US/Central')
     safe_date_time = datetime.now(central).strftime("%m%d_%H%M")  # Date and time DD-HHMM
@@ -70,7 +70,9 @@ def save_and_play_audio(audio_recorder):
         return filename
     return None
 
-def create_file(filename, prompt, response):
+def create_file(filename, prompt, response, should_save=True):
+    if not should_save:
+        return
     if filename.endswith(".txt"):
         with open(filename, 'w') as file:
             file.write(f"{prompt}\n{response}")
@@ -294,21 +296,21 @@ def divide_prompt(prompt, max_length):
     return chunks
     
 def main():
-    # Sidebar and global
+    # Sidebar and global setup
     openai.api_key = os.getenv('OPENAI_API_KEY')
-    st.set_page_config(page_title="GPT Streamlit Document Reasoner",layout="wide")
+    st.set_page_config(page_title="GPT Streamlit Document Reasoner", layout="wide")
 
     # File type for output, model choice
-    menu = ["txt", "htm", "xlsx", "csv", "md", "py"]  #619
+    menu = ["txt", "htm", "xlsx", "csv", "md", "py"]
     choice = st.sidebar.selectbox("Output File Type:", menu)
     model_choice = st.sidebar.radio("Select Model:", ('gpt-3.5-turbo', 'gpt-3.5-turbo-0301'))
-    
+
     # Audio, transcribe, GPT:
     filename = save_and_play_audio(audio_recorder)
     if filename is not None:
         transcription = transcribe_audio(openai.api_key, filename, "whisper-1")
         st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
-        filename=None # since transcription is finished next time just use the saved transcript
+        filename = None
 
     # prompt interfaces
     user_prompt = st.text_area("Enter prompts, instructions & questions:", '', height=100)
@@ -318,9 +320,15 @@ def main():
     with collength:
         max_length = st.slider("File section length for large files", min_value=1000, max_value=128000, value=12000, step=1000)
     with colupload:
-        uploaded_file = st.file_uploader("Add a file for context:", type=["pdf", "xml", "json", "xlsx","csv","html", "htm", "md", "txt"])
+        uploaded_file = st.file_uploader("Add a file for context:", type=["pdf", "xml", "json", "xlsx", "csv", "html", "htm", "md", "txt"])
+
+    should_save = st.sidebar.checkbox("💾 Save")
+
 
+
+        
     # Document section chat
+        
     document_sections = deque()
     document_responses = {}
     if uploaded_file is not None:
@@ -343,7 +351,7 @@ def main():
                     st.write(response)
                     document_responses[i] = response
                     filename = generate_filename(f"{user_prompt}_section_{i+1}", choice)
-                    create_file(filename, user_prompt, response)
+                    create_file(filename, user_prompt, response, should_save)
                     st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
 
     if st.button('💬 Chat'):
@@ -367,7 +375,7 @@ def main():
         st.write(response)
         
         filename = generate_filename(user_prompt, choice)
-        create_file(filename, user_prompt, response)
+        create_file(filename, user_prompt, response, should_save)
         st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
 
     all_files = glob.glob("*.*")
@@ -411,7 +419,7 @@ def main():
             st.write('Reasoning with your inputs...')
             response = chat_with_model(user_prompt, file_contents, model_choice)
             filename = generate_filename(file_contents, choice)
-            create_file(filename, file_contents, response)
+            create_file(filename, user_prompt, response, should_save)
 
             st.experimental_rerun()
             #st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
@@ -439,4 +447,11 @@ with st.sidebar:
             st.session_state.conversation = get_chain(vectorstore)
             st.markdown('# AI Search Index of Length:' + length + ' Created.')  # add timing
             filename = generate_filename(raw, 'txt')
-            create_file(filename, raw, '')
+            create_file(filename, raw, '', should_save)
+            #create_file(filename, raw, '')
+
+# Added "Delete All" button
+if st.sidebar.button("🗑 Delete All"):
+    for file in all_files:
+        os.remove(file)
+    st.experimental_rerun()