Create main.py

main.py

@@ -0,0 +1,69 @@
+import gradio as gr
+from core import SuperBot
+from config import APP_NAME, APP_VERSION
+from database import TemplateDB
+from utils import load_history, save_history
+
+bot = SuperBot()
+bot.history = load_history()
+
+def respond(message, history):
+    response = bot.chat(message)
+    history.append((message, response))
+    save_history(bot.history)
+    return history, ""
+
+def generate_prompt(subject, style):
+    return bot.generate_prompt(subject, style)
+
+def generate_image(prompt):
+    url = bot.call_reve(prompt)
+    return url
+
+with gr.Blocks(title=APP_NAME, theme=gr.themes.Soft()) as app:
+    gr.Markdown(f"""
+    # {APP_NAME} v{APP_VERSION}
+    Asistente sin censura • Prompts • Generación de imágenes • Multi-idioma
+    ⚠️ Responsable del uso del contenido generado
+    """)
+
+    with gr.Row():
+        api_status = gr.Textbox(label="Estado API", value="✅ SambaNova conectado" if bot else "⚠️ Sin API")
+    
+    with gr.Tabs():
+        # Tab Chat
+        with gr.TabItem("💬 Chat Inteligente"):
+            chatbot = gr.Chatbot(height=500)
+            msg = gr.Textbox(placeholder="Escribe aquí... (ej: 'crea prompt erótico' o 'genera imagen sensual')")
+            msg.submit(respond, [msg, chatbot], [chatbot, msg])
+        
+        # Tab Generador
+        with gr.TabItem("🎨 Generador de Prompts"):
+            subject = gr.Textbox(label="Sujeto/Modelo")
+            style = gr.Radio(["sensual", "profesional", "artístico", "editorial", "erótico elegante"], label="Estilo")
+            gen_btn = gr.Button("Generar Prompt")
+            prompt_out = gr.Textbox(label="Prompt listo", lines=10)
+            gen_btn.click(generate_prompt, [subject, style], prompt_out)
+            
+            with gr.Accordion("Plantillas rápidas"):
+                gr.Button("🔥 Erótico").click(lambda: TemplateDB.get("erótico"), None, prompt_out)
+                gr.Button("🏰 Fantasía").click(lambda: TemplateDB.get("fantasy"), None, prompt_out)
+                gr.Button("👻 Horror").click(lambda: TemplateDB.get("horror"), None, prompt_out)
+        
+        # Tab Imagen
+        with gr.TabItem("🖼️ Generar Imagen"):
+            img_prompt = gr.Textbox(label="Prompt para imagen")
+            img_btn = gr.Button("Generar Imagen")
+            image_out = gr.Image(label="Resultado")
+            img_btn.click(generate_image, [img_prompt], image_out)
+        
+        # Tab Ayuda
+        with gr.TabItem("ℹ️ Ayuda"):
+            gr.Markdown("""
+            **Comandos útiles:**
+            - "genera imagen [descripción]"
+            - "prompt erótico [detalles]"
+            - "crea proyecto nuevo"
+            """)
+
+app.launch(server_name="0.0.0.0", server_port=7860)