Update app.py

app.py

@@ -9,6 +9,7 @@ from datetime import datetime
 from streamlit.components.v1 import html
 from PIL import Image
 import re
+from urllib.parse import quote
 
 # Set up the Anthropic client
 client = anthropic.Anthropic(api_key=os.environ.get("ANTHROPIC_API_KEY"))
@@ -18,8 +19,11 @@ if "chat_history" not in st.session_state:
     st.session_state.chat_history = []
 
 # Function to get file download link
-def get_download_link(file_contents, file_name):
-    b64 = base64.b64encode(file_contents.encode()).decode()
+def get_download_link(file_path):
+    with open(file_path, "rb") as file:
+        contents = file.read()
+    b64 = base64.b64encode(contents).decode()
+    file_name = os.path.basename(file_path)
     return f'<a href="data:file/txt;base64,{b64}" download="{file_name}">Download {file_name}</a>'
 
 # Function to generate filename
@@ -54,6 +58,23 @@ def display_glossary_entity(k):
     links_md = ' '.join([f"[{emoji}]({url(k)})" for emoji, url in search_urls.items()])
     st.markdown(f"**{k}** <small>{links_md}</small>", unsafe_allow_html=True)
 
+# Function to create zip of files
+def create_zip_of_files(files):
+    import zipfile
+    zip_name = "all_files.zip"
+    with zipfile.ZipFile(zip_name, 'w') as zipf:
+        for file in files:
+            zipf.write(file)
+    return zip_name
+
+# Function to get zip download link
+def get_zip_download_link(zip_file):
+    with open(zip_file, 'rb') as f:
+        data = f.read()
+    b64 = base64.b64encode(data).decode()
+    href = f'<a href="data:application/zip;base64,{b64}" download="{zip_file}">Download All</a>'
+    return href
+
 # Streamlit app
 def main():
     st.title("Claude 3.5 Sonnet API Demo")