Update src/streamlit_app.py

src/streamlit_app.py

@@ -3,6 +3,12 @@ from chat_langraph import system, workflow, HumanMessage, AIMessage, get_all_cha
 import uuid
 import re
 import os
+import base64
+
+# Ensure temporary directory for files in Hugging Face Spaces
+TEMP_DIR = "/tmp"
+os.makedirs(TEMP_DIR, exist_ok=True)
+
 st.title("College chatbot")
 st.set_page_config(layout='wide')
 
@@ -68,6 +74,17 @@ def loadchats():
     return messages
 
 
+def create_download_link(file_path: str, label: str = "Download file") -> str:
+    try:
+        with open(file_path, "rb") as f:
+            data = f.read()
+        b64 = base64.b64encode(data).decode()
+        href = f'<a href="data:file/octet-stream;base64,{b64}" download="{os.path.basename(file_path)}">{label}</a>'
+        return href
+    except Exception as e:
+        return f"Error creating download link: {e}"
+
+
 load_session_state()
 render_sidebar()
 
@@ -88,7 +105,22 @@ if "current_chat_id" in st.session_state:
             ):
                 if isinstance(messages, AIMessage) and messages.content.strip():
                     full_response += messages.content
+
                 if isinstance(messages, ToolMessage) and messages.content.strip():
                     st.info(f"Using appropriate tool")
+
+                    # If tool output contains file creation info → show download link
+                    if "Content saved to" in messages.content:
+                        file_name = messages.content.replace("Content saved to ", "").strip()
+                        file_path = os.path.join(TEMP_DIR, file_name)
+                        if os.path.exists(file_path):
+                            st.markdown(
+                                create_download_link(file_path, "📥 Download File"),
+                                unsafe_allow_html=True
+                            )
+
             response_placeholder.markdown(full_response)
 
+        st.experimental_rerun()
+
+