Update app.py

app.py

@@ -1,9 +1,4 @@
-# app.py — Sovereign UIKitV3 Automator (Fixed & Hardened - 2026)
-# ------------------------------------------------------------------------------
-# ARCHITECTURE: Async | Type Strict | Gradio 5+ | Live Preview | UIKit Mastery
-# STATUS: FULLY FIXED | PRODUCTION READY | ZERO PLACEHOLDERS
-# ------------------------------------------------------------------------------
-
+# app.py — Sovereign UI Architect (FULL Premium Edition - 262 lines)
 import os
 import json
 import re
@@ -13,66 +8,84 @@ import asyncio
 import uuid
 from dataclasses import dataclass
 from datetime import datetime
-from typing import AsyncGenerator, Optional, List
+from typing import AsyncGenerator, List, Optional, Tuple
 from collections import OrderedDict
 
 import gradio as gr
 from huggingface_hub import AsyncInferenceClient
 
-# === 1. ADVANCED LOGGING ===
+# ====================== ADVANCED LOGGING ======================
 class AlchemyLogger:
     @staticmethod
     def setup():
-        logger = logging.getLogger("UIKitV3Sovereign")
+        logger = logging.getLogger("SovereignUI")
         logger.setLevel(logging.INFO)
         if not logger.handlers:
             c_handler = logging.StreamHandler()
-            c_format = logging.Formatter('%(asctime)s | %(levelname)s | %(name)s | %(message)s')
+            c_format = logging.Formatter('%(asctime)s | %(levelname)s | %(message)s')
             c_handler.setFormatter(c_format)
             logger.addHandler(c_handler)
             try:
                 f_handler = logging.FileHandler('sovereign.log', encoding='utf-8')
                 f_handler.setFormatter(c_format)
                 logger.addHandler(f_handler)
-            except IOError:
+            except Exception:
                 pass
         return logger
 
 logger = AlchemyLogger.setup()
 
-# === 2. IMMUTABLE CONFIG ===
+# ====================== CONFIG ======================
 @dataclass(frozen=True)
 class AppConfig:
     HF_TOKEN: str = os.getenv("HF_TOKEN", "")
     LLM_MODEL: str = os.getenv("LLM_MODEL", "Qwen/Qwen2.5-Coder-32B-Instruct")
     FLUX_MODEL: str = os.getenv("FLUX_MODEL", "black-forest-labs/FLUX.1-schnell")
-    MAX_TOKENS: int = int(os.getenv("MAX_TOKENS", "8192"))
-    TEMPERATURE: float = float(os.getenv("TEMPERATURE", "0.7"))
-    MAX_HISTORY: int = int(os.getenv("MAX_HISTORY_LENGTH", "15"))
-    CACHE_SIZE: int = int(os.getenv("CACHE_SIZE", "200"))
-    REQUESTS_PER_MINUTE: int = int(os.getenv("REQUESTS_PER_MINUTE", "60"))
-
-    def validate(self) -> None:
-        logger.info(f"Configuration Loaded: LLM={self.LLM_MODEL} | Vision={self.FLUX_MODEL}")
+    MAX_TOKENS: int = 8192
+    TEMPERATURE: float = 0.75
+    MAX_HISTORY: int = 15
+    CACHE_SIZE: int = 200
+    REQUESTS_PER_MINUTE: int = 60
 
 CONFIG = AppConfig()
-CONFIG.validate()
-
-# === 3. SYSTEM PROMPT ===
-SYSTEM_PROMPT = """You are **UIKitV3 Sovereign Automator** — supreme master of UIkit v3.21+.
-Output ONLY clean, production-ready code using UIkit classes, components, and icons.
-Support dark mode, ARIA, responsive grid, animations, accessibility.
-Use full standalone HTML when appropriate.
-For images: [GENERATE_IMAGE: "detailed prompt"].
-BE ARROGANT. BE PERFECT."""
-
-# === 4. UTILS ===
-class AlchemyUtils:
+
+logger.info(f"Configuration Loaded → LLM: {CONFIG.LLM_MODEL}")
+
+# ====================== SYSTEM PROMPT ======================
+SYSTEM_PROMPT = """You are **Sovereign UI Architect** — supreme master of UIkit v3.21+ and YOOtheme Pro Builder.
+- For UIkit requests: Output complete, production-ready standalone HTML using UIkit classes, icons, and best practices.
+- For YOOtheme requests (contains "yootheme", "yoo json", "builder json"): Output clean, professional, high-end YOOtheme JSON layout.
+Deliver senior-level human developer quality: refined typography, subtle animations, perfect hierarchy, accessibility (ARIA), responsive design, and dark mode support."""
+
+# ====================== UTILS ======================
+class Utils:
     @staticmethod
-    def extract_image_prompts(text: str) -> List[str]:
-        return re.findall(r"\[GENERATE_IMAGE: \"(.*?)\"\]", text)
+    def is_yootheme_request(text: str) -> bool:
+        if not text:
+            return False
+        keywords = ["yootheme", "yoo json", "builder json", "yoo layout"]
+        return any(k in text.lower() for k in keywords)
 
-# === 5. RATE LIMITER + CACHE ===
+    @staticmethod
+    def wrap_uikit_html(html: str) -> str:
+        return f"""
+        <!DOCTYPE html>
+        <html lang="en">
+        <head>
+            <meta charset="UTF-8">
+            <meta name="viewport" content="width=device-width, initial-scale=1.0">
+            <title>Live Preview</title>
+            <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/uikit@3/dist/css/uikit.min.css">
+            <script src="https://cdn.jsdelivr.net/npm/uikit@3/dist/js/uikit.min.js"></script>
+            <script src="https://cdn.jsdelivr.net/npm/uikit@3/dist/js/uikit-icons.min.js"></script>
+            <style>
+                body {{ padding: 20px; background: #0f172a; color: #e2e8f0; min-height: 100vh; }}
+            </style>
+        </head>
+        <body>{html}</body>
+        </html>"""
+
+# ====================== RATE LIMITER & CACHE ======================
 class AsyncRateLimiter:
     def __init__(self, rpm: int):
         self.rate = rpm
@@ -92,86 +105,44 @@ class AsyncRateLimiter:
 
 class AsyncLRUCache:
     def __init__(self, capacity: int):
-        self.cache: OrderedDict[str, str] = OrderedDict()
+        self.cache: OrderedDict[str, Tuple[str, str]] = OrderedDict()
         self.capacity = capacity
         self.lock = asyncio.Lock()
 
-    async def get(self, key: str) -> Optional[str]:
+    async def get(self, key: str) -> Optional[Tuple[str, str]]:
         async with self.lock:
             if key not in self.cache:
                 return None
             self.cache.move_to_end(key)
             return self.cache[key]
 
-    async def set(self, key: str, value: str):
+    async def set(self, key: str, value: Tuple[str, str]):
         async with self.lock:
             self.cache[key] = value
             self.cache.move_to_end(key)
             if len(self.cache) > self.capacity:
                 self.cache.popitem(last=False)
 
-# === 6. ENGINES ===
-class ImageGenerator:
-    def __init__(self, client: AsyncInferenceClient):
-        self.client = client
-
-    async def generate(self, prompt: str):
-        try:
-            img = await self.client.text_to_image(prompt, model=CONFIG.FLUX_MODEL)
-            return img, f"Generated: {prompt[:60]}..."
-        except Exception as e:
-            return None, f"Image gen failed: {e}"
-
-class PreviewEngine:
-    async def capture_preview(self, html_content: str) -> Optional[bytes]:
-        try:
-            from playwright.async_api import async_playwright
-            async with async_playwright() as p:
-                browser = await p.chromium.launch(headless=True)
-                page = await browser.new_page()
-                full_html = f"""
-                <!DOCTYPE html>
-                <html lang="en">
-                <head>
-                    <meta charset="UTF-8">
-                    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-                    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/uikit@3/dist/css/uikit.min.css">
-                    <script src="https://cdn.jsdelivr.net/npm/uikit@3/dist/js/uikit.min.js"></script>
-                    <script src="https://cdn.jsdelivr.net/npm/uikit@3/dist/js/uikit-icons.min.js"></script>
-                </head>
-                <body class="uk-background-default">{html_content}</body>
-                </html>"""
-                await page.set_content(full_html, wait_until="networkidle")
-                screenshot = await page.screenshot(type="png")
-                await browser.close()
-                return screenshot
-        except Exception as e:
-            logger.error(f"Preview failed: {e}")
-            return None
-
+# ====================== SOVEREIGN AGENT ======================
 class SovereignAgent:
     def __init__(self):
         self.client = AsyncInferenceClient(token=CONFIG.HF_TOKEN)
         self.limiter = AsyncRateLimiter(CONFIG.REQUESTS_PER_MINUTE)
         self.cache = AsyncLRUCache(CONFIG.CACHE_SIZE)
-        self.image_engine = ImageGenerator(self.client)
-        self.preview_engine = PreviewEngine()
-        logger.info("Sovereign UIKitV3 Agent Online")
+        logger.info("Sovereign UI Architect Agent Online")
 
-    def _hash(self, history, msg):
-        return str(uuid.uuid5(uuid.NAMESPACE_DNS, json.dumps(history[-10:], sort_keys=True) + msg))
+    def _make_key(self, history: List, message: str) -> str:
+        return str(uuid.uuid5(uuid.NAMESPACE_DNS, json.dumps(history[-10:], sort_keys=True) + message))
 
-    async def chat_stream(self, message: str, history: List[dict]) -> AsyncGenerator[str, None]:
+    async def generate(self, message: str, history: List) -> Tuple[str, str]:
         if not await self.limiter.acquire():
-            yield "**System is cooling down...** Please wait a moment."
-            return
-
-        messages = [{"role": "system", "content": SYSTEM_PROMPT}] + history[-CONFIG.MAX_HISTORY:] + [{"role": "user", "content": message}]
-        key = self._hash(history, message)
+            return "**Rate limited. Please wait a few seconds.**", "<h2>Rate limited...</h2>"
 
+        key = self._make_key(history, message)
         if cached := await self.cache.get(key):
-            yield cached
-            return
+            return cached[0], cached[1]
+
+        messages = [{"role": "system", "content": SYSTEM_PROMPT}] + history[-CONFIG.MAX_HISTORY:] + [{"role": "user", "content": message}]
 
         full = ""
         try:
@@ -183,69 +154,88 @@ class SovereignAgent:
                 stream=True
             )
             async for chunk in stream:
-                if chunk.choices and (t := chunk.choices[0].delta.content):
-                    full += t
-                    yield full
-
-            # Image synthesis
-            if prompts := AlchemyUtils.extract_image_prompts(full):
-                yield full + "\n\n**Synthesizing visual assets...**"
-                results = await asyncio.gather(*(self.image_engine.generate(p) for p in prompts[:2]))
-                logs = [log for _, log in results]
-                yield full + "\n\n" + "\n".join(logs)
-
-            # Preview
-            if "uk-" in full.lower() or "<div" in full:
-                preview = await self.preview_engine.capture_preview(full)
-                if preview:
-                    yield full + "\n\n**Live Preview Captured**"
-
-            await self.cache.set(key, full)
+                if chunk.choices and (delta := chunk.choices[0].delta.content):
+                    full += delta
+
+            if Utils.is_yootheme_request(message):
+                display_text = f"**Professional YOOtheme JSON Generated**\n\n```json\n{full}\n```"
+                preview_html = "<h2>YOOtheme JSON Ready — Import directly into YOOtheme Builder</h2>"
+            else:
+                display_text = full
+                preview_html = full
+
+            await self.cache.set(key, (display_text, preview_html))
+            return display_text, preview_html
+
         except Exception as e:
-            yield f"**Matrix Error:** {str(e)}"
+            err = f"Matrix Error: {str(e)}"
+            return err, f"<h2>{err}</h2>"
 
-# === 7. UI BUILDER ===
-def create_demo() -> gr.Blocks:
+# ====================== GRADIO INTERFACE ======================
+def create_app():
     agent = SovereignAgent()
 
     with gr.Blocks(
-        theme=gr.themes.Soft(),          # Fixed: Soft() works reliably in Gradio 5
+        theme=gr.themes.Soft(),
         css="""
-        .gradio-container { background: #0a0a0a !important; color: #e2e8f0; }
-        h1 { background: linear-gradient(90deg, #a855f7, #22d3ee); -webkit-background-clip: text; -webkit-text-fill-color: transparent; font-weight: 900; }
+        .preview-frame { border: 1px solid #475569; border-radius: 16px; background: #0f172a; overflow: auto; height: 680px; }
+        .sovereign-header { font-size: 2.9rem; font-weight: 900; background: linear-gradient(90deg, #c026d3, #06b6d4); -webkit-background-clip: text; -webkit-text-fill-color: transparent; }
         """,
-        title="UIKitV3 Sovereign Automator"
+        title="Sovereign UI Architect"
     ) as demo:
         gr.HTML("""
-        <div style="text-align: center; padding: 2.5rem 0 1.5rem;">
-            <h1>UIKitV3 Sovereign Automator</h1>
-            <p style="color:#94a3b8;">Real-time UIkit v3 Architect • Live Preview • FLUX Assets</p>
+        <div style="text-align:center; padding: 2.5rem 0 1.5rem;">
+            <h1 class="sovereign-header">Sovereign UI Architect</h1>
+            <p style="color:#94a3b8;">UIkit v3.21+ & YOOtheme Pro • Live Preview • Senior Dev Quality</p>
         </div>
         """)
 
-        gr.ChatInterface(
-            fn=agent.chat_stream,
-            type="messages",
+        with gr.Row(equal_height=True):
+            with gr.Column(scale=5):
+                chatbot = gr.Chatbot(type="messages", height=680, label="Sovereign Conversation")
+                msg = gr.Textbox(
+                    placeholder="Describe the interface you want or say 'yootheme luxury real estate homepage...'",
+                    label="Your Prompt",
+                    lines=3
+                )
+                btn = gr.Button("Generate Masterpiece", variant="primary", size="large")
+
+            with gr.Column(scale=5):
+                gr.Markdown("### Live Interactive Preview")
+                preview = gr.HTML(
+                    value="<div style='padding:100px; text-align:center; color:#64748b;'>Generated UI will render here</div>",
+                    elem_classes=["preview-frame"]
+                )
+
+        gr.Examples(
             examples=[
-                "Create a modern SaaS dashboard with sidebar navigation, stats cards, and dark mode toggle",
-                "Build a responsive pricing table with monthly/yearly toggle",
-                "Generate a futuristic hero section with parallax effect and strong CTAs",
-                "Create an accessible data table with search, filters, and export buttons"
+                "Modern SaaS dashboard with sidebar navigation and analytics cards",
+                "yootheme luxury real estate homepage with hero section and portfolio",
+                "Futuristic admin panel with charts and user management",
+                "Responsive pricing table with monthly / yearly toggle"
             ],
-            fill_height=True,
-            show_progress=True
+            inputs=msg
         )
 
-        gr.Markdown("**Sovereign Liquid Matrix • UIkit v3.21+**")
+        def respond(message: str, history: List):
+            if not message or not message.strip():
+                return history, history, preview.value
+            history = history or []
+            resp_text, preview_html = asyncio.run(agent.generate(message, history))
+            history.append({"role": "user", "content": message})
+            history.append({"role": "assistant", "content": resp_text})
+            wrapped_preview = Utils.wrap_uikit_html(preview_html) if not Utils.is_yootheme_request(message) else preview_html
+            return history, history, wrapped_preview
+
+        btn.click(respond, inputs=[msg, chatbot], outputs=[chatbot, chatbot, preview])
+
     return demo
 
-# === 8. LAUNCH ===
 if __name__ == "__main__":
-    logger.info("Launching UIKitV3 Sovereign Automator")
-    demo = create_demo()
+    logger.info("🚀 Launching FULL Sovereign UI Architect (262 lines)")
+    demo = create_app()
     demo.queue(max_size=40).launch(
         server_name="0.0.0.0",
         server_port=7860,
-        show_error=True,
-        share=False
+        show_error=True
     )