Tihsrah-CD's picture
updated
5a43d63
raw
history blame
1.35 kB
import streamlit as st
import requests
import os
def find_file(filename, directory):
for root, dirs, files in os.walk(directory):
if filename in files:
return os.path.join(root, filename)
return None
def main():
st.title('Download File from OneDrive')
search_filename = "english_vocab.pkl"
download_link = "https://upesstd-my.sharepoint.com/:u:/g/personal/500082340_stu_upes_ac_in/EYwRTq9dcTJHppgydRR-8BMBYY2BehA6jxri5rKehcSZig?e=fjAYDf"
save_filename = "classifer.joblib"
found_path = find_file(search_filename, os.getcwd())
if found_path:
st.success(f"Found {search_filename} at {found_path}")
if st.button('Download File'):
response = requests.get(download_link, allow_redirects=True)
if response.status_code == 200:
save_path = os.path.join(os.path.dirname(found_path), save_filename)
with open(save_path, 'wb') as file:
file.write(response.content)
st.success(f"File downloaded successfully and saved as {save_path}")
else:
st.error(f"Failed to download the file. Status code: {response.status_code}")
else:
st.error(f"File {search_filename} not found in the current directory or subdirectories.")
if __name__ == '__main__':
main()