Update app.py

app.py

@@ -30,9 +30,12 @@ def create_file(filename, prompt, response):
         file.write(f"<h1>Prompt:</h1> <p>{prompt}</p> <h1>Response:</h1> <p>{response}</p>")
 
 def get_table_download_link(file_path):
-    href = f'<a href="{file_path}" target="_blank">{os.path.basename(file_path)}</a>'
+    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)}">{os.path.basename(file_path)}</a>'
     return href
-
+    
 def main():
     st.title("Chat with AI")