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()