from git import Repo 
import os

import shutil
from distutils.dir_util import copy_tree

import nltk
nltk.download('punkt_tab')


git_url = os.getenv("GIT_URL")
repo_dir = "./my_number_temp"
# Specify the path to the folder
# Check if the folder exists and remove it if it does
if os.path.exists(repo_dir) and os.path.isdir(repo_dir):
    print(f"Already clone")
else:
    print(f"Folder '{repo_dir}' does not exist.")
    Repo.clone_from(git_url, repo_dir)

if os.getenv("FORCE", 'NO') != 'NO':
    shutil.rmtree(repo_dir)
    print("Force clone")
    Repo.clone_from(git_url, repo_dir)

# Specify the folder you want to move and the current folder path
source_folder = "./my_number_temp/hnrecommender/"
destination_folder = f"{os.getcwd()}/hnrecommender"  # Gets the path of the current 

print(destination_folder)
# Check if the destination folder exists and remove it if it does
# if not os.path.exists(destination_folder):
os.makedirs(destination_folder,exist_ok=True)
# Copy the source folder to the destination
copy_tree(source_folder, destination_folder)

import streamlit as st

from hnrecommender import recommend_hacker_news

# Streamlit UI
st.title("HNews Recommendation")

# Text input for the user to enter a query
user_bio = st.text_area("Enter the user bio:", height=100)

# Submit button
if st.button("Submit"):
    if user_bio:
        # Show spinner
        with st.spinner("Fetching articles... Please wait."):
            articles = recommend_hacker_news(user_bio, 500)

        # Display the results after processing
        st.success("Here are the articles recommended for you:")
        for story in articles:
            title = story["title"] if "title" in story else "No title article"
            url = story["url"] if "url" in story else "HN article"
            st.write(f"[{title}]({url})")
    else:
        st.error("Please enter an user bio.")