import gradio as gr
import shutil
import os

FORMATS = {
    "📝 Word Document (.docx)": ".docx",
    "📄 PowerPoint (.pptx)": ".pptx",
    "📊 Excel Spreadsheet (.xlsx)": ".xlsx",
    "🖼️ JPEG Image (.jpg)": ".jpg",
    "🖼️ PNG Image (.png)": ".png",
    "📄 PDF (.pdf)": ".pdf",
    "📄 Text File (.txt)": ".txt",
    "🎬 MP4 Video (.mp4)": ".mp4",
    "🎵 MP3 Audio (.mp3)": ".mp3",
    "📦 RAR Archive (.rar)": ".rar",
    "📦 7-Zip Archive (.7z)": ".7z",
    "📦 TAR Archive (.tar)": ".tar",
    "🌐 HTML File (.html)": ".html",
    "📋 CSV File (.csv)": ".csv",
    "🔧 JSON File (.json)": ".json",
}

def convert_zip(zip_file, format_label):
    if zip_file is None:
        return None, "❌ Please upload a ZIP file first.", ""

    ext = FORMATS.get(format_label)
    if not ext:
        return None, "❌ Please select a valid target format.", ""

    original_name = os.path.basename(zip_file.name)
    base_name = os.path.splitext(original_name)[0]
    output_filename = base_name + ext
    output_path = os.path.join("/tmp", output_filename)

    print(f"Converting: {original_name} → {output_filename}")
    shutil.copy(zip_file.name, output_path)

    ext_label = ext.lstrip('.')
    prompt = (
        f"I have changed the ZIP file to {ext_label} format. "
        f"Please change the file extension back to .zip before processing."
    )

    status = (
        f"✅ Done! `{original_name}` → `{output_filename}`\n\n"
        f"📋 Download the file below, then copy the prompt and paste it alongside your upload."
    )

    return output_path, status, prompt


theme = gr.themes.Soft(
    primary_hue="red",
    neutral_hue="gray",
    font=[gr.themes.GoogleFont("DM Sans"), "sans-serif"],
)

css = """
body, .gradio-container { font-family: 'DM Sans', sans-serif !important; }
#title { text-align: center; margin-bottom: 0.2em; font-size: 2em; }
#subtitle { text-align: center; color: #64748b; font-size: 0.95em; margin-bottom: 1.5em; }
#github_link { text-align: center; margin-bottom: 0.5em; }
.prompt-box textarea { font-family: 'DM Sans', sans-serif !important; font-size: 0.9em !important; }
footer { display: none !important; }
"""

with gr.Blocks(title="ZipSlip — Upload ZIPs Anywhere") as demo:

    gr.Markdown("# 📦 ZipSlip", elem_id="title")
    gr.Markdown(
        "**https://github.com/SanshruthR/ZipSlip/**",
        elem_id="github_link",
    )
    gr.Markdown(
        "Rename your ZIP so you can upload it to AI tools that block ZIP files — "
        "Kimi and other web interfaces. "
        "A ready-to-paste prompt is generated so the AI knows to treat it as a ZIP.",
        elem_id="subtitle",
    )

    with gr.Row():
        with gr.Column(scale=1):
            zip_input = gr.File(
                label="📂 Upload your ZIP file",
                file_types=[".zip"],
                type="filepath",
            )
            format_dropdown = gr.Dropdown(
                choices=list(FORMATS.keys()),
                value="📝 Word Document (.docx)",
                label="🔀 Disguise as format",
            )
            convert_btn = gr.Button("🚀 Disguise & Generate Prompt", variant="primary", size="lg")

        with gr.Column(scale=1):
            status_box = gr.Markdown(label="Status")
            file_output = gr.File(label="⬇️ Download disguised file")
            prompt_output = gr.Textbox(
                label="📋 Paste this prompt alongside your upload",
                lines=3,
                interactive=True,
                elem_classes=["prompt-box"],
                placeholder="Your prompt will appear here after converting…",
            )

    gr.Markdown(
        "---\n"
        "**How to use:** Upload your ZIP → choose a disguise format → download the renamed file "
        "→ upload it to the AI tool → paste the prompt so the AI knows to treat it as a ZIP.\n\n"
        "> ⚠️ This tool only renames the file — no actual conversion happens. "
        "The AI needs to rename it back to `.zip` before extracting."
    )

    convert_btn.click(
        fn=convert_zip,
        inputs=[zip_input, format_dropdown],
        outputs=[file_output, status_box, prompt_output],
    )

if __name__ == "__main__":
    demo.launch(theme=theme, css=css)