Update backupapp.py

backupapp.py

@@ -16,11 +16,8 @@ openai.api_key = os.getenv('OPENAI_KEY')
 st.set_page_config(
     page_title="GPT Streamlit Document Reasoner",
     layout="wide")
-# st.title("GPT Chat with Optional File Context - Talk to your data!")
 
-# Output options sidebar menu
-# st.sidebar.title("Output Options")
-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":
@@ -30,32 +27,36 @@ 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.")
 
-# 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)
+max_length = st.sidebar.slider("Max document length", min_value=1000, max_value=32000, value=2000, step=1000)
 
-# Truncate document
 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)
     return response['choices'][0]['message']['content']
 
-def generate_filename(prompt):
+def generate_filename(prompt, file_type):
     central = pytz.timezone('US/Central')
-    safe_date_time = datetime.now(central).strftime("%m%d_%I_%M_%p")  
-    safe_prompt = "".join(x for x in prompt if x.isalnum())[:30]
-    return f"{safe_date_time}_{safe_prompt}.txt"
-    
+    safe_date_time = datetime.now(central).strftime("%m%d_%I%M")  
+    safe_prompt = "".join(x for x in prompt if x.isalnum())[:28]
+    return f"{safe_date_time}_{safe_prompt}.{file_type}"
+
 def create_file(filename, prompt, response):
-    with open(filename, 'w') as file:
-        file.write(f"<h1>Prompt:</h1> <p>{prompt}</p> <h1>Response:</h1> <p>{response}</p>")
+    if filename.endswith(".txt"):
+        with open(filename, 'w') as file:
+            file.write(f"Prompt:\n{prompt}\nResponse:\n{response}")
+    elif filename.endswith(".htm"):
+        with open(filename, 'w') as file:
+            file.write(f"<h1>Prompt:</h1> <p>{prompt}</p> <h1>Response:</h1> <p>{response}</p>")
+    elif filename.endswith(".md"):
+        with open(filename, 'w') as file:
+            file.write(f"# Prompt:\n{prompt}\n# Response:\n{response}")
 
 def get_table_download_link_old(file_path):
     with open(file_path, 'r') as file:
@@ -132,19 +133,22 @@ def main():
         st.write('Response:')
         st.write(response)
         
-        filename = generate_filename(user_prompt)
+        filename = generate_filename(user_prompt, choice)
         create_file(filename, user_prompt, response)
         st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
     
     if len(file_content) > 0:
         st.markdown(f"**File Content Added:**\n{file_content}")
     
-    htm_files = glob.glob("*.txt")
-    for file in htm_files:
-        st.sidebar.markdown(get_table_download_link(file), unsafe_allow_html=True)
-        if st.sidebar.button(f"🗑 Delete {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
+        with col1:
+            st.markdown(get_table_download_link(file), unsafe_allow_html=True)
+        with col2:
+            if st.button("🗑", key=file):
+                os.remove(file)
+                st.experimental_rerun()
 
 if __name__ == "__main__":
     main()