Update app.py

app.py

@@ -16,7 +16,8 @@ openai.api_key = os.getenv('OPENAI_KEY')
 st.set_page_config(
     page_title="GPT Streamlit Document Reasoner",
     layout="wide")
-menu = ["txt", "htm", "md"]
+
+menu = ["txt", "htm", "md", "py"]
 choice = st.sidebar.selectbox("Choose output file type to save results", menu)
 choicePrefix = "Output and download file set to "
 if choice == "txt":
@@ -26,16 +27,15 @@ elif choice == "htm":
 elif choice == "md":
      st.sidebar.write(choicePrefix + "Markdown.")
 elif choice == "py":
-     st.sidebar.write(choicePrefix + "Python AI UI/UX")
+     st.sidebar.write(choicePrefix + "Python Code.")
 
-max_length = st.sidebar.slider("Max document length", min_value=1000, max_value=32000, value=3000, step=1000)
+max_length = st.sidebar.slider("Max document length", min_value=1000, max_value=32000, value=2000, step=1000)
 
 def truncate_document(document, length):
     return document[:length]
 
 def chat_with_model(prompts):
     model = "gpt-3.5-turbo"
-    #model = "gpt-4-32k" # 32k tokens between prompt and inference tokens
     conversation = [{'role': 'system', 'content': 'You are a helpful assistant.'}]
     conversation.extend([{'role': 'user', 'content': prompt} for prompt in prompts])
     response = openai.ChatCompletion.create(model=model, messages=conversation)
@@ -140,13 +140,6 @@ def main():
     if len(file_content) > 0:
         st.markdown(f"**File Content Added:**\n{file_content}")
     
-    #all_files = glob.glob("*.txt") + glob.glob("*.htm") + glob.glob("*.md")
-    #for file in all_files:
-    #    st.sidebar.markdown(get_table_download_link(file), unsafe_allow_html=True)
-    #    if st.sidebar.button("🗑", key=file):
-    #        os.remove(file)
-    #        st.experimental_rerun()
-
     all_files = glob.glob("*.txt") + glob.glob("*.htm") + glob.glob("*.md")
     for file in all_files:
         col1, col2 = st.sidebar.columns([4,1])  # adjust the ratio as needed
@@ -157,6 +150,5 @@ def main():
                 os.remove(file)
                 st.experimental_rerun()
 
-
 if __name__ == "__main__":
     main()