import gradio as gr
import subprocess
import os

def setup_fooocus():
    try:
        # Directorio de trabajo
        work_dir = '/home/user/app/my_project/Fooocus'
        
        # Asegúrate de que el directorio de trabajo exista
        os.makedirs(work_dir, exist_ok=True)
        
        # Cambia el directorio de trabajo a la ruta deseada
        os.chdir(work_dir)
        
        # Instalar dependencias y clonar el repositorio
        subprocess.run(['pip', 'install', 'pygit2==1.15.1'], check=True)
        subprocess.run(['git', 'clone', 'https://github.com/lllyasviel/Fooocus.git'], check=True)
        os.chdir('Fooocus')
        
        # Ejecutar el script
        result = subprocess.run(['python', 'entry_with_update.py', '--always-high-vram'], capture_output=True, text=True)
        
        # Devolver la salida del script
        return result.stdout

    except Exception as e:
        return f"Error: {str(e)}"

# Crear la interfaz de Gradio
demo = gr.Interface(fn=setup_fooocus, inputs=[], outputs="text")

# Lanzar la aplicación Gradio sin share=True
demo.launch()