update

app.py

@@ -2,7 +2,8 @@ import streamlit as st
 import snscrape.modules.twitter as sntwitter
 import pandas as pd
 import plotly.express as px
-import os
+import base64
+
 
 st.set_page_config(page_title="Scraping Twitter")
 
@@ -69,9 +70,17 @@ if st.button('Scrape Tweets'):
     st.write(f'Tweets involving @{selected_username}:')
     st.write(tweets_df[tweets_df['Username'] == selected_username])
 
-    # Save tweets as CSV file in data folder
-    if not os.path.exists("data"):
-        os.mkdir("data")
-    file_name = f"data/tweets_{query.replace(' ', '_')}.csv"
-    tweets_df.to_csv(file_name, index=False)
-    st.write(f"Tweets saved as {file_name}")
+# Download CSV
+    if len(tweets_df) > 0:
+        csv = tweets_df.to_csv(index=False)
+        b64 = base64.b64encode(csv.encode()).decode()
+        filename = f'tweets_{query.replace(" ", "_")}.csv'
+        href = f'<a href="data:file/csv;base64,{b64}" download="{filename}">Download CSV</a>'
+        st.markdown(href, unsafe_allow_html=True)
+
+    if len(tweets_df) > 0:
+        txt = "\n\n".join(tweets_df['Text'].tolist())
+        b64 = base64.b64encode(txt.encode()).decode()
+        txt_filename = f"{query}_tweets.txt"
+        txt_href = f'<a href="data:file/txt;base64,{b64}" download="{txt_filename}">Download TXT</a>'
+        st.markdown(txt_href, unsafe_allow_html=True)

requirements.txt

@@ -3,3 +3,5 @@ snscrape
 plotly
 seaborn
 pandas
+base64
+