import gradio as gr
from g4f.client import Client

def generate_image(prompt):
    try:
        client = Client()
        response = client.images.generate(
            model="flux-realism",
            prompt=prompt,
            # Можете добавить дополнительные параметры
        )
        
        # Получаем URL изображения
        image_url = response.data[0].url
        
        return image_url
    except Exception as e:
        return f"Ошибка при генерации изображения: {str(e)}"

# Создаем интерфейс Gradio
def create_image_generator():
    with gr.Blocks() as demo:
        gr.Markdown("# Генератор изображений с помощью AI")
        
        with gr.Row():
            # Текстовое поле для ввода промпта
            prompt_input = gr.Textbox(label="Опишите изображение")
            
            # Кнопка генерации
            generate_btn = gr.Button("Сгенерировать изображение")
        
        # Область для отображения сгенерированного изображения
        output_image = gr.Image(label="Сгенерированное изображение")
        
        # Привязываем функцию генерации к кнопке
        generate_btn.click(
            fn=generate_image, 
            inputs=prompt_input, 
            outputs=output_image
        )
    
    return demo

# Запуск приложения
if __name__ == "__main__":
    demo = create_image_generator()
    demo.launch(
        share=True,  # Создает публичную ссылку
        debug=True   # Включает режим отладки
    )