Update backupapp.py

backupapp.py

@@ -1,19 +1,18 @@
 import streamlit as st
 import openai
 import os
-
-from streamlit_chat import message
+import base64
+import glob
+from datetime import datetime
 from dotenv import load_dotenv
 from openai import ChatCompletion
 
 load_dotenv()
 
 openai.api_key = os.getenv('OPENAI_KEY')
-# keys are here: https://platform.openai.com/auth/callback?code=ReZ4izEw0DwkUKrHR-Opxr5AMMgo9SojxC9pNHQUcjD6M&state=OGZFNDJmLlJGNlIwOUxlakpXZkVFfjNxNy02ZlFtLWN4eUcuOXJobXouSQ%3D%3D#
 
-# Define a function to chat with the model
 def chat_with_model(prompts):
-    model = "gpt-3.5-turbo"  # change this to the model you're using
+    model = "gpt-3.5-turbo"  
 
     conversation = [{'role': 'system', 'content': 'You are a helpful assistant.'}]
     conversation.extend([{'role': 'user', 'content': prompt} for prompt in prompts])
@@ -21,21 +20,22 @@ def chat_with_model(prompts):
     response = openai.ChatCompletion.create(model=model, messages=conversation)
     return response['choices'][0]['message']['content']
 
-load_dotenv('api_key.env')
-
-def generate_response(prompt):
-    completion=openai.Completion.create(
-        engine='text-davinci-003',
-        prompt=prompt,
-        max_tokens=1024,
-        n=1,
-        stop=None,
-        temperature=0.6,
-    )
-    message=completion.choices[0].text
-    return message
-
-# Streamlit App
+def generate_filename(prompt):
+    safe_date_time = datetime.now().strftime("%Y_%m_%d_%H_%M_%S")
+    safe_prompt = "".join(x for x in prompt if x.isalnum())[:50]
+    return f"{safe_date_time}_{safe_prompt}.md"
+
+def create_file(filename, prompt, response):
+    with open(filename, 'w') as file:
+        file.write(f"Prompt: {prompt}\n\nResponse: {response}")
+
+def get_table_download_link(file_path):
+    with open(file_path, 'r') as file:
+        data = file.read()
+    b64 = base64.b64encode(data.encode()).decode()  
+    href = f'<a href="data:file/txt;base64,{b64}" download="{os.path.basename(file_path)}">Download Response File</a>'
+    return href
+
 def main():
     st.title("Chat with AI")
 
@@ -54,5 +54,14 @@ def main():
         st.write('Response:')
         st.write(response)
 
+        filename = generate_filename(user_prompt)
+        create_file(filename, user_prompt, response)
+
+        st.markdown(get_table_download_link(filename), unsafe_allow_html=True)
+
+    md_files = glob.glob("*.md")
+    for file in md_files:
+        st.markdown(get_table_download_link(file), unsafe_allow_html=True)
+
 if __name__ == "__main__":
-    main()
+    main()