Update app.py

app.py

@@ -47,15 +47,22 @@ def chat_with_model(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"
-    
+    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,17 +139,17 @@ 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:
+    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(f"🗑 Delete {file}"):
+        if st.sidebar.button("🗑", key=file):
             os.remove(file)
             st.experimental_rerun()