app.py

import gradio as gr
import subprocess
import os

def run_katana(url):
    output_file = "urls.txt"
    command = f"katana -u {url} > {output_file}"
    subprocess.run(command, shell=True, check=True)
    return output_file

# Ensure the /tmp/flagged directory exists
os.makedirs("/tmp/flagged", exist_ok=True)

# Gradio interface
interface = gr.Interface(
    fn=run_katana,
    inputs="text",
    outputs="file",
    title="Katana URL Crawler",
    description="Enter a URL to crawl with Katana.",
    flagging_dir="/tmp/flagged"  # Specify a writable directory for flagging
)

interface.launch(server_name="0.0.0.0", server_port=7860)