Update app.py

app.py

@@ -16,7 +16,6 @@ from bs4 import BeautifulSoup
 from collections import deque
 from audio_recorder_streamlit import audio_recorder
 
-
 def generate_filename(prompt, file_type):
     central = pytz.timezone('US/Central')
     safe_date_time = datetime.now(central).strftime("%m%d_%I%M")  
@@ -135,23 +134,6 @@ def read_file_content(file,max_length):
         return ""
 
 
-# Sidebar and global
-openai.api_key = os.getenv('OPENAI_KEY')
-st.set_page_config(page_title="GPT Streamlit Document Reasoner",layout="wide")
-menu = ["htm", "txt", "xlsx", "csv", "md", "py"]  #619
-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.write(transcription)
-    gptOutput = chat_with_model(transcription, '') # *************************************
-    filename = generate_filename(transcription, choice)
-    create_file(filename, transcription, gptOutput)
-    st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
-
 
 def chat_with_file_contents(prompt, file_content):
     conversation = [{'role': 'system', 'content': 'You are a helpful assistant.'}]
@@ -160,6 +142,7 @@ def chat_with_file_contents(prompt, file_content):
         conversation.append({'role': 'assistant', 'content': file_content})
     response = openai.ChatCompletion.create(model=model_choice, messages=conversation)
     return response['choices'][0]['message']['content']
+        
 
 # Sidebar and global
 openai.api_key = os.getenv('OPENAI_KEY')
@@ -173,16 +156,18 @@ filename = save_and_play_audio(audio_recorder)
 if filename is not None:
     transcription = transcribe_audio(openai.api_key, filename, "whisper-1")
     st.write(transcription)
-    gptOutput = chat_with_model(transcription, '') 
+    gptOutput = chat_with_model(transcription, '') # *************************************
     filename = generate_filename(transcription, choice)
     create_file(filename, transcription, gptOutput)
     st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
 
+
 def main():
     user_prompt = st.text_area("Enter prompts, instructions & questions:", '', height=100)
 
     collength, colupload = st.columns([2,3])  # adjust the ratio as needed
     with collength:
+        #max_length = 12000 - optimal for gpt35 turbo. 2x=24000 for gpt4.  8x=96000 for gpt4-32k.
         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=["xml", "json", "xlsx","csv","html", "htm", "md", "txt"])
@@ -208,17 +193,27 @@ def main():
             else:
                 if st.button(f"Chat about Section {i+1}"):
                     st.write('Reasoning with your inputs...')
-                    response = chat_with_model(user_prompt, section)
+                    response = chat_with_model(user_prompt, section) # *************************************
                     st.write('Response:')
                     st.write(response)
                     document_responses[i] = response
-                    filename = generate_filename(f"{user_prompt}_Section_{i+1}", choice)
+                    filename = generate_filename(f"{user_prompt}_section_{i+1}", choice)
                     create_file(filename, user_prompt, response)
                     st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
 
-    # ... document_responses logic remains the same
-    
-    all_files = [file for file in glob.glob("*.{}".format(choice))]
+    if st.button('💬 Chat'):
+        st.write('Reasoning with your inputs...')
+        response = chat_with_model(user_prompt, ''.join(list(document_sections))) # *************************************
+        st.write('Response:')
+        st.write(response)
+        
+        filename = generate_filename(user_prompt, choice)
+        create_file(filename, user_prompt, response)
+        st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
+
+    all_files = glob.glob("*.*")
+    all_files = [file for file in all_files if len(os.path.splitext(file)[0]) >= 20]  # 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
     
     for file in all_files:
         col1, col2, col3 = st.sidebar.columns([5,1,1])  # adjust the ratio as needed
@@ -234,6 +229,7 @@ def main():
                     response = chat_with_file_contents(user_prompt, file_contents)
                     st.write('Response:')
                     st.write(response)
+                    
                     filename = generate_filename(user_prompt, choice)
                     create_file(filename, user_prompt, response)
                     st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
@@ -241,6 +237,6 @@ def main():
             if st.button("🗑", key="delete_"+file):
                 os.remove(file)
                 st.experimental_rerun()
-
+                
 if __name__ == "__main__":
-    main()
+    main()