import streamlit as st
import requests
import os
import shutil
import zipfile
from github import Github
from git import Repo

def download_github_repo(url, local_path):
    # Clone the repository
    Repo.clone_from(url, local_path)

def create_zip_file(source_dir, output_filename):
    shutil.make_archive(output_filename, 'zip', source_dir)

def check_repo_exists(g, repo_name):
    try:
        g.get_repo(repo_name)
        return True
    except:
        return False

def create_repo(g, repo_name):
    user = g.get_user()
    user.create_repo(repo_name.split('/')[-1])

def delete_repo(g, repo_name):
    repo = g.get_repo(repo_name)
    repo.delete()

def push_to_github(local_path, repo_url, github_token):
    repo = Repo(local_path)
    origin = repo.create_remote('origin', repo_url)
    origin.push(refspec='{}:{}'.format('master', 'master'))

def main():
    st.title("GitHub Repository Cloner and Uploader")

    # Input for source repository
    source_repo = st.text_input("Source GitHub Repository URL", 
                                value="https://github.com/AaronCWacker/AIExamples-8-24-Streamlit/")

    # Input for target repository
    target_repo = st.text_input("Target GitHub Repository URL", 
                                value="https://github.com/AaronCWacker/AIExample-Clone/")

    # GitHub token input
    github_token = st.text_input("GitHub Personal Access Token", type="password")

    if st.button("Clone and Upload"):
        with st.spinner("Processing..."):
            # Download the source repository
            local_path = "./temp_repo"
            download_github_repo(source_repo, local_path)

            # Create a zip file
            zip_filename = "repo_contents"
            create_zip_file(local_path, zip_filename)

            # Check if the target repository exists
            g = Github(github_token)
            repo_name = '/'.join(target_repo.split('/')[-2:])
            repo_exists = check_repo_exists(g, repo_name)

            if repo_exists:
                overwrite = st.radio("Repository already exists. Overwrite?", ("Yes", "No"))
                if overwrite == "Yes":
                    delete_repo(g, repo_name)
                    create_repo(g, repo_name)
                else:
                    st.warning("Operation cancelled. Target repository already exists.")
                    return
            else:
                create_repo(g, repo_name)

            # Push the contents to the new repository
            push_to_github(local_path, target_repo, github_token)

            # Clean up
            shutil.rmtree(local_path)
            os.remove(zip_filename + ".zip")

            st.success("Repository cloned and uploaded successfully!")

if __name__ == "__main__":
    main()