Update app.py

app.py

@@ -92,6 +92,22 @@ def create_knowledge_base(data):
 
     return qa_chain
 
+def get_shareable_link(file_data, file_name, file_type):
+    """
+    Generates a shareable link for the file using a base64 data URL.
+
+    Args:
+        file_data: The file data in bytes.
+        file_name: The name of the file.
+        file_type: The MIME type of the file (e.g., 'text/csv', 'application/vnd.ms-excel').
+
+    Returns:
+        A base64-encoded data URL for the file.
+    """
+    b64 = base64.b64encode(file_data).decode()
+    return f"data:{file_type};base64,{b64}"
+    
+
 def main():
     st.sidebar.title("Quantilytix Grant Scraper")
     st.sidebar.image("logoqb.jpeg", use_container_width=True)
@@ -129,6 +145,8 @@ def main():
             b64 = base64.b64encode(csv_data).decode()
             download_link = f"<a href='data:application/vnd.ms-excel;base64,{b64}' download='grants.csv'>Download CSV</a>"
             st.sidebar.markdown(download_link, unsafe_allow_html=True)
+            shareable_link = get_shareable_link(result, file_name, file_type)
+            
         elif selected_format == "Excel":
             excel_data = convert_to_excel(result)
             b64 = base64.b64encode(excel_data).decode()
@@ -137,13 +155,15 @@ def main():
 
         st.dataframe(result['grants'])
 
+        
+
         if st.sidebar.button("Load as Knowledge Base"):
             st.session_state.qa_chain = create_knowledge_base(result)
             st.session_state.chat_interface_active = True
 
     if "chat_interface_active" in st.session_state and st.session_state.chat_interface_active:
-        st.image("logoqb.jpeg", use_container_width=True)
-        st.heading("Chat Interface Loaded. Start asking questions about the grants!")
+        st.image("logoqb.jpeg", width=100, height=100)
+        st.header("Chat Interface Loaded. Start asking questions about the grants!")
 
         query = st.text_input("Ask a question about the grants:", key="chat_input")
         if query: