Update app.py

app.py

@@ -2,123 +2,409 @@ import os
 import requests
 import gradio as gr
 
+API_BASE_URL = os.getenv("SOLRICKS_API_BASE_URL", "https://api.solricks.com")
+API_KEY = os.getenv("SOLRICKS_SPACE_API_KEY", "")
+
+API_ENDPOINT = f"{API_BASE_URL}/api/workflows/visual-ai/run"
+
+ALLOWED_RESOLUTIONS = ["1024x1024", "768x1344", "1344x768"]
+ALLOWED_STYLES = ["Portrait", "Realistic", "Cinematic", "Product"]
+
 custom_css = """
+:root {
+    --sr-bg: #f7f9ff;
+    --sr-card: rgba(255, 255, 255, 0.92);
+    --sr-card-strong: #ffffff;
+    --sr-text: #091225;
+    --sr-muted: #667085;
+    --sr-soft: #eef2ff;
+    --sr-border: rgba(90, 85, 255, 0.16);
+    --sr-border-strong: rgba(90, 85, 255, 0.24);
+    --sr-purple: #6d5dfc;
+    --sr-blue: #2557ff;
+    --sr-shadow: 0 18px 48px rgba(15, 23, 42, 0.08);
+    --sr-shadow-soft: 0 12px 30px rgba(15, 23, 42, 0.055);
+}
+
 .gradio-container {
-    max-width: 1180px !important;
+    max-width: 1520px !important;
     margin: 0 auto !important;
+    background:
+        radial-gradient(circle at 10% 8%, rgba(109, 93, 252, 0.14), transparent 30%),
+        radial-gradient(circle at 88% 12%, rgba(37, 87, 255, 0.10), transparent 30%),
+        linear-gradient(180deg, #fbfcff 0%, var(--sr-bg) 100%) !important;
+    color: var(--sr-text) !important;
+}
+
+#solricks-shell {
+    padding: 34px 22px 18px;
 }
 
 #solricks-hero {
-    background:
-        radial-gradient(circle at 80% 20%, rgba(96, 165, 250, 0.28), transparent 35%),
-        radial-gradient(circle at 10% 0%, rgba(124, 58, 237, 0.25), transparent 35%),
-        linear-gradient(135deg, #070814 0%, #111827 45%, #1e1b4b 100%);
-    padding: 30px;
-    border-radius: 22px;
-    margin-bottom: 22px;
-    color: white;
-    border: 1px solid rgba(255,255,255,0.12);
-    box-shadow: 0 18px 55px rgba(15, 23, 42, 0.22);
+    margin-bottom: 18px;
 }
 
 #solricks-kicker {
-    display: inline-flex;
-    align-items: center;
-    gap: 8px;
-    padding: 7px 12px;
-    border-radius: 999px;
-    background: rgba(255,255,255,0.10);
-    color: rgba(255,255,255,0.82);
+    color: #111a32;
     font-size: 12px;
-    font-weight: 700;
-    letter-spacing: 0.08em;
+    font-weight: 900;
+    letter-spacing: 0.30em;
     text-transform: uppercase;
-    border: 1px solid rgba(255,255,255,0.14);
-    margin-bottom: 16px;
+    margin-bottom: 12px;
 }
 
 #solricks-hero h1 {
-    margin: 0 0 10px 0;
-    color: white;
-    font-size: 38px;
-    line-height: 1.05;
-    font-weight: 900;
-    letter-spacing: -0.04em;
+    margin: 0;
+    color: #081126;
+    font-size: clamp(40px, 5vw, 66px);
+    line-height: 0.98;
+    font-weight: 950;
+    letter-spacing: -0.055em;
+}
+
+#solricks-hero h1 span {
+    background: linear-gradient(90deg, #7c3aed 0%, #4f46e5 45%, #2557ff 100%);
+    -webkit-background-clip: text;
+    background-clip: text;
+    color: transparent;
 }
 
 #solricks-hero p {
-    margin: 0;
+    margin: 14px 0 0 0;
     max-width: 760px;
-    font-size: 15px;
+    color: #34405c;
+    font-size: 16px;
     line-height: 1.65;
-    color: rgba(255,255,255,0.78);
 }
 
-#solricks-note-row {
+#solricks-badges {
     display: flex;
     flex-wrap: wrap;
     gap: 10px;
     margin-top: 18px;
 }
 
-.solricks-pill {
+.solricks-badge {
     display: inline-flex;
     align-items: center;
-    padding: 8px 12px;
-    border-radius: 999px;
-    background: rgba(255,255,255,0.10);
-    color: rgba(255,255,255,0.88);
+    gap: 10px;
+    min-height: 46px;
+    padding: 9px 14px;
+    border-radius: 15px;
+    border: 1px solid var(--sr-border);
+    background: rgba(255, 255, 255, 0.74);
+    box-shadow: var(--sr-shadow-soft);
+}
+
+.solricks-badge-icon {
+    width: 28px;
+    height: 28px;
+    display: inline-flex;
+    align-items: center;
+    justify-content: center;
+    border-radius: 10px;
+    background: linear-gradient(135deg, rgba(124, 58, 237, 0.16), rgba(37, 87, 255, 0.12));
+    color: #5b4bff;
+    font-size: 15px;
+}
+
+.solricks-badge strong {
+    display: block;
     font-size: 13px;
-    font-weight: 700;
-    border: 1px solid rgba(255,255,255,0.14);
+    line-height: 1.1;
+    color: #111827;
+    font-weight: 900;
+}
+
+.solricks-badge small {
+    display: block;
+    margin-top: 3px;
+    font-size: 11px;
+    color: #7b8498;
+    font-weight: 650;
 }
 
-#solricks-panel-note {
-    padding: 14px 16px;
-    border-radius: 16px;
-    background: rgba(79, 70, 229, 0.08);
-    border: 1px solid rgba(79, 70, 229, 0.16);
+.solricks-card {
+    border: 1px solid var(--sr-border) !important;
+    border-radius: 22px !important;
+    background: var(--sr-card) !important;
+    box-shadow: var(--sr-shadow) !important;
+    overflow: hidden !important;
+}
+
+.solricks-card + .solricks-card {
+    margin-top: 14px !important;
+}
+
+.solricks-panel {
+    padding: 18px !important;
+}
+
+.solricks-card-title {
+    display: flex;
+    align-items: center;
+    justify-content: space-between;
+    gap: 12px;
+    margin-bottom: 10px;
+    color: #10172a;
+    font-size: 15px;
+    font-weight: 900;
+}
+
+.solricks-card-title-left {
+    display: inline-flex;
+    align-items: center;
+    gap: 8px;
+}
+
+.solricks-step {
+    width: 22px;
+    height: 22px;
+    display: inline-flex;
+    align-items: center;
+    justify-content: center;
+    border-radius: 999px;
+    background: rgba(109, 93, 252, 0.17);
+    color: #5b4bff;
+    font-size: 12px;
+    font-weight: 950;
+}
+
+.solricks-clear {
+    color: #667085;
     font-size: 13px;
-    line-height: 1.55;
-    margin-bottom: 12px;
+    font-weight: 800;
 }
 
-#solricks-footer {
-    margin-top: 22px;
+.solricks-field-label {
+    display: flex;
+    align-items: center;
+    gap: 8px;
+    margin: 2px 0 9px;
+    color: #10172a;
+    font-size: 14px;
+    font-weight: 900;
+}
+
+textarea,
+input,
+select {
+    border-radius: 15px !important;
+}
+
+textarea {
+    min-height: 132px !important;
+}
+
+.wrap,
+.block,
+.form,
+.input-container,
+.output-class {
+    border-radius: 18px !important;
+}
+
+.solricks-generate-button button,
+button.primary {
+    min-height: 54px !important;
+    border: 0 !important;
+    border-radius: 16px !important;
+    color: #ffffff !important;
+    background: linear-gradient(90deg, #7c3aed 0%, #5b4bff 46%, #2557ff 100%) !important;
+    box-shadow: 0 14px 32px rgba(79, 70, 229, 0.30) !important;
+    font-size: 16px !important;
+    font-weight: 950 !important;
+    letter-spacing: -0.01em;
+}
+
+.solricks-generate-button button:hover,
+button.primary:hover {
+    transform: translateY(-1px);
+    filter: brightness(1.03);
+}
+
+#solricks-status textarea {
+    min-height: 54px !important;
+    height: 54px !important;
+}
+
+#solricks-status label {
+    display: none !important;
+}
+
+#solricks-image-panel {
+    min-height: 566px !important;
+}
+
+#solricks-preview-header {
     padding: 18px 20px;
-    border-radius: 18px;
-    background: rgba(15, 23, 42, 0.04);
+    border-bottom: 1px solid var(--sr-border);
+    display: flex;
+    align-items: center;
+    justify-content: space-between;
+    gap: 12px;
+    background: rgba(255, 255, 255, 0.68);
+}
+
+#solricks-preview-title {
+    display: inline-flex;
+    align-items: center;
+    gap: 10px;
+    color: #10172a;
+    font-size: 18px;
+    font-weight: 950;
+}
+
+#solricks-preview-title span {
+    width: 30px;
+    height: 30px;
+    display: inline-flex;
+    align-items: center;
+    justify-content: center;
+    border-radius: 11px;
+    background: rgba(109, 93, 252, 0.14);
+    color: #5b4bff;
+}
+
+#solricks-download-ghost {
+    padding: 9px 13px;
+    border-radius: 12px;
     border: 1px solid rgba(15, 23, 42, 0.08);
+    color: #98a2b3;
     font-size: 13px;
-    color: rgba(15, 23, 42, 0.72);
+    font-weight: 850;
+    background: rgba(255, 255, 255, 0.70);
 }
 
-#solricks-footer a {
-    font-weight: 700;
-    text-decoration: none;
+#solricks-output-image,
+#solricks-output-image > div {
+    min-height: 500px !important;
 }
 
-button.primary {
-    border-radius: 14px !important;
-    font-weight: 800 !important;
+#solricks-output-image img {
+    border-radius: 0 0 22px 22px !important;
+    object-fit: contain !important;
+}
+
+#solricks-output-image .empty,
+#solricks-output-image .icon-wrap {
+    color: #7b8498 !important;
+}
+
+#solricks-examples-wrap {
+    margin-top: 18px;
 }
 
-textarea, input, select {
-    border-radius: 12px !important;
+#solricks-examples-head {
+    display: flex;
+    justify-content: space-between;
+    gap: 14px;
+    margin: 0 0 8px;
+    color: #10172a;
+    font-size: 15px;
+    font-weight: 950;
 }
 
-.svelte-1ipelgc, .wrap {
+#solricks-examples-head small {
+    color: #8a94a8;
+    font-size: 12px;
+    font-weight: 750;
+}
+
+#solricks-examples-wrap table {
+    overflow: hidden !important;
     border-radius: 16px !important;
 }
-"""
 
-API_BASE_URL = os.getenv("SOLRICKS_API_BASE_URL", "https://api.solricks.com")
-API_KEY = os.getenv("SOLRICKS_SPACE_API_KEY", "")
+#solricks-footer {
+    margin-top: 16px;
+    padding: 18px 22px;
+    border-radius: 20px;
+    border: 1px solid var(--sr-border);
+    background: rgba(255, 255, 255, 0.84);
+    box-shadow: var(--sr-shadow-soft);
+    display: flex;
+    align-items: center;
+    justify-content: space-between;
+    gap: 24px;
+    flex-wrap: wrap;
+}
 
-API_ENDPOINT = f"{API_BASE_URL}/api/workflows/visual-ai/run"
+#solricks-footer-brand {
+    display: inline-flex;
+    align-items: center;
+    gap: 12px;
+    color: #0b1224;
+    font-size: 24px;
+    font-weight: 950;
+    letter-spacing: -0.035em;
+}
 
-ALLOWED_RESOLUTIONS = ["1024x1024", "768x1344", "1344x768"]
-ALLOWED_STYLES = ["Portrait", "Realistic", "Cinematic", "Product"]
+#solricks-logo-mark {
+    width: 42px;
+    height: 42px;
+    display: inline-flex;
+    align-items: center;
+    justify-content: center;
+    border-radius: 14px;
+    color: #5b4bff;
+    background: linear-gradient(135deg, rgba(124, 58, 237, 0.16), rgba(37, 87, 255, 0.12));
+    border: 1px solid rgba(109, 93, 252, 0.20);
+}
+
+#solricks-footer p {
+    margin: 0;
+    max-width: 720px;
+    color: #3d4660;
+    font-size: 14px;
+    line-height: 1.55;
+}
+
+#solricks-links {
+    display: flex;
+    align-items: center;
+    gap: 16px;
+    flex-wrap: wrap;
+}
+
+#solricks-links a {
+    color: #2557d6;
+    font-size: 14px;
+    font-weight: 850;
+    text-decoration: none;
+}
+
+#solricks-links a:hover {
+    text-decoration: underline;
+}
+
+footer {
+    opacity: 0.78;
+}
+
+@media (max-width: 900px) {
+    #solricks-shell {
+        padding: 24px 10px 14px;
+    }
+
+    #solricks-hero h1 {
+        font-size: 40px;
+    }
+
+    #solricks-image-panel {
+        min-height: 430px !important;
+    }
+
+    #solricks-output-image,
+    #solricks-output-image > div {
+        min-height: 360px !important;
+    }
+
+    #solricks-footer {
+        align-items: flex-start;
+    }
+}
+"""
 
 
 def generate_image(prompt, style, resolution):
@@ -198,86 +484,189 @@ with gr.Blocks(
     theme=theme,
     css=custom_css,
 ) as demo:
-    gr.Markdown(
-    """
-    <div id="solricks-hero">
-        <div id="solricks-kicker">SOLRICKS · VISUAL AI PREVIEW</div>
-        <h1>SOLRICKS Image Generator</h1>
-        <p>Generate a free preview image using the Solricks visual AI backend.</p>
-        <div id="solricks-note-row">
-            <span class="solricks-pill">Rate-limited</span>
-            <span class="solricks-pill">Preview generations</span>
-        </div>
-    </div>
-    """
-    )
+    with gr.Column(elem_id="solricks-shell"):
+        gr.HTML(
+            """
+            <section id="solricks-hero">
+                <div id="solricks-kicker">SOLRICKS · VISUAL AI PREVIEW</div>
+                <h1><span>SOLRICKS</span> Image Generator</h1>
+                <p>
+                    Generate a free preview image using the Solricks visual AI backend.
+                    Create clean visual concepts with simple controls for style and resolution.
+                </p>
+
+                <div id="solricks-badges">
+                    <div class="solricks-badge">
+                        <span class="solricks-badge-icon">✦</span>
+                        <div><strong>Free Preview</strong><small>Preview generations</small></div>
+                    </div>
+                    <div class="solricks-badge">
+                        <span class="solricks-badge-icon">⏱</span>
+                        <div><strong>Rate Limited</strong><small>Limited daily usage</small></div>
+                    </div>
+                    <div class="solricks-badge">
+                        <span class="solricks-badge-icon">⚙</span>
+                        <div><strong>Visual AI Backend</strong><small>Powered by Solricks</small></div>
+                    </div>
+                </div>
+            </section>
+            """
+        )
 
-    with gr.Row():
-        with gr.Column(scale=5):
-            prompt = gr.Textbox(
-                label="Prompt",
-                placeholder="Describe the image you want to generate...",
-                lines=6,
-                max_lines=8,
-            )
-
-            with gr.Row():
-                style = gr.Dropdown(
-                    label="Style",
-                    choices=ALLOWED_STYLES,
-                    value="Portrait",
-                )
-
-                resolution = gr.Dropdown(
-                    label="Resolution",
-                    choices=ALLOWED_RESOLUTIONS,
-                    value="1024x1024",
-                )
-
-            generate_button = gr.Button("Generate Image", variant="primary")
-
-            status = gr.Textbox(
-                label="Status",
-                interactive=False,
-                placeholder="Generation status will appear here...",
-            )
-
-        with gr.Column(scale=6):
-            output_image = gr.Image(
-                label="Generated Image",
-                height=460,
-            )
-
-    gr.Examples(
-        examples=[
-            [
-                "cinematic close-up portrait of a futuristic android, dark premium lighting, realistic details",
-                "Portrait",
-                "768x1344",
-            ],
-            [
-                "realistic AI robot in a dark premium technology lab, cinematic lighting, blue accent lights",
-                "Cinematic",
-                "1344x768",
+        with gr.Row(equal_height=True):
+            with gr.Column(scale=5, min_width=360):
+                with gr.Group(elem_classes="solricks-card"):
+                    with gr.Column(elem_classes="solricks-panel"):
+                        gr.HTML(
+                            """
+                            <div class="solricks-card-title">
+                                <div class="solricks-card-title-left">
+                                    <span class="solricks-step">1</span>
+                                    <span>Prompt</span>
+                                </div>
+                                <span class="solricks-clear">✦ Clear</span>
+                            </div>
+                            """
+                        )
+
+                        prompt = gr.Textbox(
+                            label=None,
+                            placeholder="Describe the image you want to generate...",
+                            lines=5,
+                            max_lines=7,
+                            max_length=500,
+                            show_label=False,
+                        )
+
+                        with gr.Row():
+                            with gr.Column():
+                                gr.HTML(
+                                    """
+                                    <div class="solricks-field-label">
+                                        <span class="solricks-step">2</span>
+                                        <span>Style</span>
+                                    </div>
+                                    """
+                                )
+                                style = gr.Dropdown(
+                                    label=None,
+                                    choices=ALLOWED_STYLES,
+                                    value="Portrait",
+                                    show_label=False,
+                                )
+
+                            with gr.Column():
+                                gr.HTML(
+                                    """
+                                    <div class="solricks-field-label">
+                                        <span class="solricks-step">3</span>
+                                        <span>Resolution</span>
+                                    </div>
+                                    """
+                                )
+                                resolution = gr.Dropdown(
+                                    label=None,
+                                    choices=ALLOWED_RESOLUTIONS,
+                                    value="1024x1024",
+                                    show_label=False,
+                                )
+
+                        generate_button = gr.Button(
+                            "✨ Generate Image",
+                            variant="primary",
+                            elem_classes="solricks-generate-button",
+                        )
+
+                with gr.Group(elem_classes="solricks-card"):
+                    with gr.Column(elem_classes="solricks-panel"):
+                        gr.HTML(
+                            """
+                            <div class="solricks-card-title">
+                                <div class="solricks-card-title-left">
+                                    <span class="solricks-step">4</span>
+                                    <span>Status</span>
+                                </div>
+                            </div>
+                            """
+                        )
+
+                        status = gr.Textbox(
+                            label=None,
+                            value="Ready to generate. Your generation status will appear here.",
+                            interactive=False,
+                            show_label=False,
+                            elem_id="solricks-status",
+                        )
+
+            with gr.Column(scale=6, min_width=420):
+                with gr.Group(elem_classes="solricks-card", elem_id="solricks-image-panel"):
+                    gr.HTML(
+                        """
+                        <div id="solricks-preview-header">
+                            <div id="solricks-preview-title">
+                                <span>▧</span>
+                                Generated Image
+                            </div>
+                            <div id="solricks-download-ghost">Download</div>
+                        </div>
+                        """
+                    )
+
+                    output_image = gr.Image(
+                        label=None,
+                        show_label=False,
+                        height=500,
+                        elem_id="solricks-output-image",
+                    )
+
+        gr.HTML(
+            """
+            <div id="solricks-examples-wrap">
+                <div id="solricks-examples-head">
+                    <span>☷ Examples</span>
+                    <small>Click any example to use the prompt</small>
+                </div>
+            </div>
+            """
+        )
+
+        gr.Examples(
+            examples=[
+                [
+                    "cinematic close-up portrait of a futuristic android, dark premium lighting, realistic details",
+                    "Portrait",
+                    "768x1344",
+                ],
+                [
+                    "realistic AI robot in a dark premium technology lab, cinematic lighting, blue accent lights",
+                    "Cinematic",
+                    "1344x768",
+                ],
             ],
-        ],
-        inputs=[prompt, style, resolution],
-    )
+            inputs=[prompt, style, resolution],
+        )
 
-    gr.Markdown(
-        """
-<div id="solricks-footer">
-<strong>SOLRICKS</strong> develops solution bricks for AI-powered systems, visual AI workflows,
-ComfyUI tools, LoRA resources, and GPU automation.
-<br><br>
-<a href="https://solricks.com/" target="_blank">Website</a>
-&nbsp;·&nbsp;
-<a href="https://github.com/SOLRICKS" target="_blank">GitHub</a>
-&nbsp;·&nbsp;
-<a href="https://civitai.com/user/Solricks" target="_blank">Civitai</a>
-</div>
-"""
-    )
+        gr.HTML(
+            """
+            <div id="solricks-footer">
+                <div id="solricks-footer-brand">
+                    <span id="solricks-logo-mark">◇</span>
+                    SOLRICKS
+                </div>
+
+                <p>
+                    SOLRICKS develops solution bricks for AI-powered systems, visual AI workflows,
+                    ComfyUI tools, LoRA resources, and GPU automation.
+                </p>
+
+                <div id="solricks-links">
+                    <a href="https://solricks.com/" target="_blank">Website ↗</a>
+                    <a href="https://github.com/SOLRICKS" target="_blank">GitHub ↗</a>
+                    <a href="https://civitai.com/user/Solricks" target="_blank">Civitai ↗</a>
+                </div>
+            </div>
+            """
+        )
 
     generate_button.click(
         fn=generate_image,
@@ -285,4 +674,4 @@ ComfyUI tools, LoRA resources, and GPU automation.
         outputs=[output_image, status],
     )
 
-demo.queue(max_size=10).launch()
+demo.queue(max_size=10).launch()