Update app.py

app.py

@@ -11,49 +11,32 @@ from datetime import datetime
 from openai import ChatCompletion
 from xml.etree import ElementTree as ET
 from bs4 import BeautifulSoup
-#from streamlit.theme import Theme
 
 openai.api_key = os.getenv('OPENAI_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",
-#    theme=Theme(
-#        primary_color="#F63366",
-#        secondary_background_color="#F0F2F6",
-#        text_color="#262730",
-#        font="sans serif",
-#    ),
-)
+    layout="wide")
 st.title("Chat with AI")
 
-# Create a sidebar with menus
-# st.sidebar.title("Menu")
-# menu = ["Option 1", "Option 2", "Option 3"]
-# choice = st.sidebar.selectbox("Choose an option", menu)
-
-# if choice == "Option 1":
-#     st.sidebar.write("You selected Option 1")
-#elif choice == "Option 2":
-#     st.sidebar.write("You selected Option 2")
-#elif choice == "Option 3":
-#    st.sidebar.write("You selected Option 3")
-
-# Create a slider in the sidebar
-max_length = st.sidebar.slider(
-    "Max document length", min_value=3000, max_value=24000, value=3000, step=1000
-)
+# Output options sidebar menu
+st.sidebar.title("Output Options")
+menu = ["txt", "htm", "md"]
+choice = st.sidebar.selectbox("Choose an output file type to save your prompt and results", menu)
+choicePrefix = "Output and download file set to "
+if choice == "txt":
+     st.sidebar.write(choicePrefix + "Text file.")
+elif choice == "htm":
+     st.sidebar.write(choicePrefix + "HTML5.")
+elif choice == "md":
+     st.sidebar.write(choicePrefix + "Markdown.")
+
+# sidebar slider for file input length to include in inference blocks
+max_length = st.sidebar.slider("Max document length", min_value=1000, max_value=32000, value=3000, step=1000)
 
 # Truncate document
 def truncate_document(document, length):
     return document[:length]
 
-# Assume you have a document called my_document
-# my_document = 'your long string here'
-# truncated_document = truncate_document(my_document, max_length)
-# st.write(f"Truncated document: {truncated_document}")
-    
 def chat_with_model(prompts):
     model = "gpt-3.5-turbo"
     #model = "gpt-4-32k" # 32k tokens between prompt and inference tokens
@@ -108,7 +91,6 @@ def CompressXML(xml_text):
     #return ET.tostring(root, encoding='unicode', method="xml")
     return ET.tostring(root, encoding='unicode', method="xml")[:max_length]  
     
-
 def read_file_content(file):
     if file.type == "application/json":
         content = json.load(file)
@@ -131,10 +113,8 @@ def read_file_content(file):
         return ""
 
 def main():
-
     prompts = ['']
     file_content = ""
-
     user_prompt = st.text_area("Your question:", '', height=120)
     uploaded_file = st.file_uploader("Choose a file", type=["xml", "json", "html", "htm", "md", "txt"])