UI_VCS.py

# **Installs & imports**
"""

!pip install gradio -q
!pip install html -q

# **Installs & Imports**
from difflib import Differ
from pathlib import Path
import time

import tempfile
import atexit

import os
from io import BytesIO

import gradio as gr
import re
import shlex
import html

import subprocess
import shutil

# For images
from PIL import Image
import numpy as np

"""# **Build custom theme**"""

custom_theme = gr.themes.Ocean(
    primary_hue="indigo",
    secondary_hue="yellow",
    neutral_hue="indigo",
    text_size="lg",
    radius_size="lg",
    font=['IBM Plex Sans', 'ui-sans-serif', 'system-ui', gr.themes.GoogleFont('sans-serif')],
    font_mono=['Inter', 'ui-monospace', gr.themes.GoogleFont('Consolas'), 'monospace'],
).set(
    background_fill_secondary='*secondary_50',
    background_fill_secondary_dark='*neutral_950',
    border_color_accent='*primary_50',
    border_color_primary='*neutral_300',
    color_accent='*primary_300',
    color_accent_soft_dark='*neutral_500',
    checkbox_label_background_fill='*color_accent_soft',
    button_large_radius='*radius_sm',
    button_small_radius='*radius_sm',
    button_primary_background_fill='linear-gradient(120deg, *primary_500 0%, *primary_300 60%, *primary_400 100%)',
    button_primary_background_fill_hover='radial-gradient(circle, *secondary_400 0%, *primary_300 60%, *primary_300 100%)'
)

"""# **Functions**

## Global constants file paths
"""

def create_tmp_file() -> str:
    tmp_file = tempfile.NamedTemporaryFile(mode="w+", delete=False)
    tmp_file.close()
    return tmp_file.name

def write_tmp_file(content: str, file_path: str):
    with open(file_path, "w") as file:
        file.write(content)

def read_tmp_file(file_path: str) -> str:
    with open(file_path, "r") as file:
        return file.read()

def cleanup_tmp_files():
    os.remove(CREATOR_NAME_FILE)
    os.remove(CREATOR_EMAIL_FILE)
    os.remove(PROJECT_NAME_FILE)

CREATOR_NAME_FILE = create_tmp_file()
CREATOR_EMAIL_FILE = create_tmp_file()
PROJECT_NAME_FILE = create_tmp_file()

# **Define UI functions**
# For texts
def diff_texts(text1, text2):
    d = Differ()
    return [
        (token[2:], token[0] if token[0] != " " else None)
        for token in d.compare(text1, text2)
    ]

def download_file():
    return [gr.UploadButton(visible=True), gr.DownloadButton(visible=False)]

"""## Create project"""

def validate_email(email: str, max_length: int=50):
    if not email or len(email) > max_length:
        return False
    email_regex = r'^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$'
    return re.match(email_regex, email) is not None

def validate_project_name(project_name: str, max_length: int=50):
    if not project_name or len(project_name) > max_length:
        return False
    return True

def validate_creator_name(creator_name: str, max_length: int=50):
    if not creator_name or len(creator_name) > max_length:
        return False
    return True


def run_create_project_script(project_name: str, creator_name: str, creator_email):
    if validate_project_name(project_name):
        project_name = project_name.replace(" ", "_")
    else:
        return "Empty/Invalid project name"
    if not creator_name:
        return "Empty/Invalid creator name"
    if not validate_email(creator_email):
        return "Empty/Invalid email address"

    project_name_clean = shlex.quote(html.escape(project_name))
    creator_name_clean = shlex.quote(html.escape(creator_name))
    creator_email_clean = shlex.quote(html.escape(creator_email))

    write_tmp_file(project_name_clean, PROJECT_NAME_FILE)
    write_tmp_file(creator_name_clean, CREATOR_NAME_FILE)
    write_tmp_file(creator_email_clean, CREATOR_EMAIL_FILE)

    arguments = ["./", project_name_clean, creator_name_clean, creator_email_clean]
    result = subprocess.run(["C:\\Program Files\\Git\\bin\\bash.exe", "./vcs_scripts/create_project.sh"] + arguments, capture_output=True, text=True, shell=False)
    return result.stdout

"""## Upload project"""

def validate_uploaded_text(uploaded_text: str):
    max_length=10000
    if uploaded_text is not None and uploaded_text != "" and len(uploaded_text) < max_length:
        return True
    return False

def validate_commit_message(commit_message: str):
    max_length=1000
    if commit_message is not None and len(commit_message) < max_length:
        return True
    return False

def validate_filepath(filepath: str):
    if filepath is not None and os.path.exists(filepath):
        return True
    return False

def move_file_to_project(filepath: str, project_path: str):
    if filepath == "" or filepath is None:
        return False
    if not os.path.exists(filepath):
        return False
    name = Path(filepath).name
    destination = os.path.join(project_path, name) # Create destination path: target_dir/filename
    shutil.move(filepath, destination)
    return os.path.exists(destination)

def run_upload_project_script(project_path: str, filepath: str):
    filename = Path(filepath).name
    if move_file_to_project(filepath, project_path):
        commit_message = f"Uploaded '{filename}' to '{project_path}' project"
        if validate_commit_message(commit_message):
            arguments = [os.getcwd(), project_path, commit_message]
            !sed -i 's/\r$//' ./vcs_scripts/upload2.sh
            result = subprocess.run(["C:\\Program Files\\Git\\bin\\bash.exe", "./vcs_scripts/upload2.sh"] + arguments, capture_output=True, text=True, shell=False)
            return True, f"{result.stdout}"
        else:
            return False, f"Something went wrong with commit message: {commit_message}"
    return False, "Please create a project first"



def upload_text(uploaded_text: str):
    project_path = read_tmp_file(PROJECT_NAME_FILE)
    if project_path == "":
        return gr.DownloadButton(visible=False), gr.Textbox(value="Please create a project first.", visible=True)
    if not validate_uploaded_text(uploaded_text.get("text", "")):
        return gr.DownloadButton(visible=False), gr.Textbox(value="Empty/Invalid argument entered.", visible=True)

    # Sanitize text input
    text = shlex.quote(html.escape(uploaded_text.get("text", "")))
    commit_message_clean = shlex.quote(html.escape("Uploaded "))

    filename = uploaded_text.get("name", "untitled.txt")
    with open(filename, "w", encoding="utf-8") as file:
        file.write(uploaded_text.get("text", ""))
    status, message = run_upload_project_script(project_path, os.path.join(os.getcwd(), filename))
    if status:
        return gr.DownloadButton(value=filename, label=f"Download {filename}", visible=True), gr.Textbox(value=message, visible=True)
    return gr.DownloadButton(visible=False), gr.Textbox(value=message, visible=True)

def upload_file(filepath: str):
    project_name = read_tmp_file(PROJECT_NAME_FILE)
    if project_name == "":
        return gr.DownloadButton(visible=False), gr.Textbox(value="Please create a project first.", visible=True)
    filename = Path(filepath.name).name
    project_path = os.path.join(os.getcwd(), project_name)
    status, message = run_upload_project_script(project_path, filepath)
    if status:
        return gr.DownloadButton(value=filename, label=f"Download {filename}", visible=True), gr.Textbox(value=message, visible=True) # gr.Textbox(value=message, visible=True)
    return gr.DownloadButton(visible=False), gr.Textbox(value=message, visible=True)

# For image uploads
def sleep(img_dict):
    time.sleep(5)
    return [img_dict["background"], img_dict["layers"][0], img_dict["layers"][1], img_dict["composite"]]

def show_image(img_dict):
    return img_dict["composite"]

def upload_image(img_dict):
    output_path = "output_image.png"
    image = img_dict["composite"]
    image = Image.fromarray(image)

    # Save the image to the output path
    image.save(output_path)
    return f"Image saved to {output_path}"

def run_upload_project_script(project_path: str, filepath: str):
    filename = Path(filepath).name
    if move_file_to_project(filepath, project_path):
        commit_message = f"Uploaded '{filename}' to '{project_path}' project."
        if validate_commit_message(commit_message):
            arguments = [os.getcwd(), project_path, commit_message]
            result = subprocess.run(["C:\\Program Files\\Git\\bin\\bash.exe", "./vcs_scripts/upload2.sh"] + arguments, capture_output=True, text=True, check=True, shell=False)#, shell=True)
            return True, f"Automatic commit message - {commit_message}\n{result.stdout}"
        else:
            return False, f"Something went wrong with commit message: {commit_message}"
    return False, "Please create a project first"

"""## Recreate project"""

def run_recreate_project_script(project_name: str, new_project_name: str):
    if validate_project_name(new_project_name):
        new_project_name = new_project_name.replace(" ", "_")
        new_project_name_clean = shlex.quote(html.escape(new_project_name))
    else:
        return "Empty/Invalid new project name"
    if validate_project_name(project_name):
        if os.path.isdir(project_name):
            arguments = ["./vcs_scripts", "./", project_name, new_project_name_clean]
            result = subprocess.run(["C:\\Program Files\\Git\\bin\\bash.exe", "./vcs_scripts/recreate.sh"] + arguments, capture_output=True, text=True, shell=False)
            return result.stdout
        elif os.path.isdir(project_name + ".git"): # If user forget to include .git part in entered remote project name
            arguments = ["./vcs_scripts", "./", project_name + ".git", new_project_name_clean]
            result = subprocess.run(["C:\\Program Files\\Git\\bin\\bash.exe", "./vcs_scripts/recreate.sh"] + arguments, capture_output=True, text=True, shell=False)
            return result.stdout
        else:
            return f"Project '{project_name}' does not exist."
    else:
        return f"Empty/Invalid project name {project_name}"

"""## Download project"""

def run_download_script(project_name: str):
    if validate_project_name(project_name):
        project_path = os.path.join(os.getcwd(), project_name)
        if os.path.isdir(project_path):
            arguments = ["./", project_name]
            result = subprocess.run(["C:\\Program Files\\Git\\bin\\bash.exe", "./vcs_scripts/download.sh"] + arguments, capture_output=True, text=True, shell=False)
            return result.stdout
        elif os.path.isdir(project_path + ".git"): # If user forget to include .git part in entered remote project name
            arguments = ["./", project_name + ".git"]
            result = subprocess.run(["C:\\Program Files\\Git\\bin\\bash.exe", "./vcs_scripts/download.sh"] + arguments, capture_output=True, text=True, shell=False)
            return result.stdout
        else:
            return f"Project '{project_name}' does not exist in path."
    else:
        return f"Empty/Invalid project name {project_name}"

"""## Revert project"""

def run_revert_script(project_name: str, revert_hash: str):
    if validate_project_name(project_name):
        project_path = os.path.join(os.getcwd(), project_name)
        if os.path.isdir(project_path):
            arguments = ["./vcs_script", "./", project_name, revert_hash]
            result = subprocess.run(["C:\\Program Files\\Git\\bin\\bash.exe", "./vcs_scripts/revert2.sh"] + arguments, capture_output=True, text=True, shell=False)
            return result.stdout
        elif os.path.isdir(project_path + ".git"): # If user forget to include .git part in entered project name
            arguments = ["./vcs_script", "./", project_name + ".git", revert_hash]
            result = subprocess.run(["C:\\Program Files\\Git\\bin\\bash.exe", "./vcs_scripts/revert2.sh"] + arguments, capture_output=True, text=True, shell=False)
            return result.stdout
        else:
            return f"Project '{project_name}' does not exist in path."
    else:
        return f"Empty/Invalid project name {project_name}"

"""# **Build layout**"""

os.chmod("./vcs_scripts/grant_permissions.sh", 0o755)
result = subprocess.run(["C:\\Program Files\\Git\\bin\\bash.exe", "./vcs_scripts/grant_permissions.sh"], capture_output=True, text=True, shell=False)
print(result.stdout)

with gr.Blocks(theme=custom_theme, title="Version Control") as demo:
    gr.Markdown("<h1 style='text-align: center;'>Your project</h1>")
    with gr.Tabs():
        with gr.Tab("Create Empty Project"):
            with gr.Row():
                project_name = gr.Textbox(label="Project name", placeholder="my_creation, my-creation, or myCreation")
            with gr.Row():
                creator_name = gr.Textbox(label="Creator name", placeholder="Victor Frankenstein")
                creator_email = gr.Textbox(label="Creator email", placeholder="example@gmail.com")
            with gr.Row():
                submit_btn = gr.Button("Create project", variant="primary")
            with gr.Row():
                output = gr.Textbox(label="status")
            submit_btn.click(fn=run_create_project_script, inputs=[project_name, creator_name, creator_email], outputs=output)

        with gr.Tab("Text Upload"):
            with gr.Group():
                with gr.Accordion(label="Use basic text editor...", open=False):
                    text_input = gr.MultimodalTextbox(
                        label="(max 10000 characters)",
                        sources=["microphone"],
                        file_types=[".txt"],
                        placeholder="Enter your text here or speak...",
                        submit_btn="Upload",
                        lines=8
                    )
            with gr.Group():
                file_upload = gr.File(
                            label="Upload a text file",
                            file_types=["text"],
                            file_count="single",
                            interactive=True
                )

            #commit_textbox = gr.Textbox(label="commit message", placeholder="Commit message must be under 1000 chars", visible=True, interactive=True)

            file_output = gr.DownloadButton(label="Download File", visible=False, variant="primary")
            download_button = gr.DownloadButton("Download the file", visible=False, variant="primary")
            status_textbox = gr.Textbox(label="status", visible=False, interactive=False)

            text_input.submit(fn=upload_text, inputs=[text_input], outputs=[file_output, status_textbox])
            file_upload.upload(upload_file, [file_upload], [download_button, status_textbox])
            download_button.click(download_file, None, [file_upload, download_button])

        with gr.Tab("Image Upload"):
            with gr.Row():
                img_editor = gr.ImageEditor(
                    label="Image uploader & editor",
                    type="numpy",
                    crop_size="16:8",
                    scale=4,
                    brush=gr.Brush(color_mode="defaults", colors=[
                        "#FFADAD",
                        "#FFD6A5",
                        "#FDFFB6",
                        "#CAFFBF",
                        "#9BF6FF",
                        "#A0C4FF",
                        "#BDB2FF",
                        "#FFC6FF",

                        # Neutral colors:
                        "#FFFFFF",  # White
                        "#808080",  # Gray
                        "#000000",  # Black

                        # Skin tones and browns:
                        "#F5D0C5",  # Peach
                        "#E0A98A",  # Sand
                        "#F1C6A0",  # Gold
                        "#EDA268",  # Tan
                        "#D0743A",  # Bronze
                        "#6B3F1F",  # Chocolate
                        "#3C0000"   # Mahogany
                        ]
                    ),
                    canvas_size=(1920,1080)
                )

            with gr.Row():
                img_preview = gr.Image(label="Result image")
            with gr.Row():
                n_upload = gr.Number(0, label="Number of upload events", step=1, interactive=False)
                n_change = gr.Number(0, label="Number of change events", step=1, interactive=False)
            with gr.Row():
                upload_btn = gr.Button("Upload image")
            with gr.Row():
                output_text = gr.Textbox(label="Status", interactive=False)
            img_editor.upload(lambda x: x + 1, outputs=n_upload, inputs=n_upload)
            img_editor.change(lambda x: x + 1, outputs=n_change, inputs=n_change)
            img_editor.change(show_image, outputs=img_preview, inputs=img_editor, show_progress="hidden")


            upload_btn.click(fn=upload_image, inputs=img_editor, outputs=output_text)

        with gr.Tab("Recreate"):
            with gr.Row():
                project_name = gr.Textbox(label="Remote project you want to recreate",
                           interactive=True)
                new_project_name = gr.Textbox(label="Your new project name",
                           interactive=True)
            with gr.Row():
                submit_btn = gr.Button("Recreate project", variant="primary")
            with gr.Row():
                recreate_output = gr.Textbox(label="status", interactive=False)
                submit_btn.click(fn=run_recreate_project_script, inputs=[project_name, new_project_name], outputs=recreate_output)
        """
        with gr.Tab("Download"):
            with gr.Row():
                project_name = gr.Textbox(label="Project you want to download",
                           interactive=True)
            with gr.Row():
                submit_btn = gr.Button("Download any new changes", variant="primary")
            with gr.Row():
                download_output = gr.Textbox(label="status", interactive=False)
                submit_btn.click(fn=run_download_script, inputs=[project_name], outputs=download_output)
        """
        with gr.Tab("Compare"):
            with gr.Row():
                gr.Markdown("<h1>Compare version differences</h1>")
            with gr.Row():
                original_version = gr.Textbox(
                    label="Original",
                    info="Initial text",
                    lines=4,
                    value="Blasted as thou wert, my agony was still superior to thine.",
                    interactive=False,
                )
                new_version = gr.Textbox(
                    label="Uploaded",
                    info="New changes",
                    lines=4,
                    value="Blasted as thou (Victor Frankenstein) wert, my (the creature's) agony was still superior to his (because creature is ugly and has no friends)",
                    interactive=False,
                )

            compare_btn = gr.Button("Compare differences", variant="primary")
            with gr.Row():
                result = gr.HighlightedText(
                    label="Diff",
                    combine_adjacent=True,
                    show_legend=True,
                    color_map={"+": "blue", "-": "red"}
                )
            compare_btn.click(fn=diff_texts, inputs=[original_version, new_version], outputs=result)
        with gr.Tab("Revert"):
            with gr.Row():
                project_name = gr.Textbox(label="Project name", interactive=True)
                revert_hash = gr.Textbox(label="Revert back to...",
                                        info="hash value",
                                        interactive=True
                )
            with gr.Row():
                submit_btn = gr.Button("Revert", variant="primary")
            with gr.Row():
                revert_output = gr.Textbox(label="status", interactive=False)
                submit_btn.click(fn=run_revert_script, inputs=[project_name, revert_hash], outputs=revert_output)


if __name__ == "__main__":
    demo.launch(debug=True, share=True)
    atexit.register(cleanup_tmp_files)