import gradio as gr

def main():
    # Laden des Flux-Modells
    demo = gr.load("models/enhanceaiteam/Flux-Uncensored-V2")
    
    # Anpassung der Benutzeroberfläche
    with gr.Blocks() as flux_interface:
        gr.Markdown("# Flux Image Generator 🖼️")
        gr.Markdown("Generieren Sie hochwertige Bilder mit Flux AI")
        
        with gr.Row():
            # Der geladene Flux-Komponente wird automatisch hinzugefügt
            demo.render()
        
        gr.Markdown("""
        ### Tipps zur Nutzung:
        - Geben Sie detaillierte Beschreibungen ein
        - Experimentieren Sie mit verschiedenen Stilen
        - Mehr Details führen zu präziseren Ergebnissen
        """)
    
    # Interface starten
    flux_interface.launch(
        debug=True,   # Detaillierte Fehlermeldungen
        show_error=True  # Zeige Fehler im Interface
    )

if __name__ == "__main__":
    main()