Update app.py

app.py

@@ -2,6 +2,42 @@ import os
 import requests
 import gradio as gr
 
+custom_css = """
+#solricks-hero {
+    background: linear-gradient(135deg, #1e1b4b 0%, #312e81 45%, #2563eb 100%);
+    padding: 24px;
+    border-radius: 18px;
+    margin-bottom: 18px;
+    color: white;
+    box-shadow: 0 10px 30px rgba(37, 99, 235, 0.18);
+}
+
+#solricks-hero h1 {
+    margin: 0 0 8px 0;
+    color: white;
+    font-size: 32px;
+    font-weight: 800;
+}
+
+#solricks-hero p {
+    margin: 0;
+    font-size: 15px;
+    color: rgba(255,255,255,0.92);
+}
+
+#solricks-note {
+    display: inline-block;
+    margin-top: 14px;
+    padding: 7px 12px;
+    border-radius: 999px;
+    background: rgba(255,255,255,0.14);
+    color: white;
+    font-size: 13px;
+    font-weight: 600;
+    border: 1px solid rgba(255,255,255,0.18);
+}
+"""
+
 API_BASE_URL = os.getenv("SOLRICKS_API_BASE_URL", "https://api.solricks.com")
 API_KEY = os.getenv("SOLRICKS_SPACE_API_KEY", "")
 
@@ -77,15 +113,18 @@ def generate_image(prompt, style, resolution):
         return None, f"Error: {error}"
 
 
-with gr.Blocks(title="SOLRICKS Image Generator") as demo:
+with gr.Blocks(
+    title="SOLRICKS Image Generator",
+    css=custom_css,
+) as demo:
     gr.Markdown(
         """
-        # SOLRICKS Image Generator
-
-        Free visual AI preview powered by the Solricks backend.
-
-        This public demo is rate-limited and intended for short preview generations.
-        """
+<div id="solricks-hero">
+    <h1>SOLRICKS Image Generator</h1>
+    <p>Free visual AI preview powered by the Solricks backend.</p>
+    <span id="solricks-note">Public demo · Rate-limited · Preview generations</span>
+</div>
+"""
     )
 
     prompt = gr.Textbox(
@@ -108,11 +147,10 @@ with gr.Blocks(title="SOLRICKS Image Generator") as demo:
             value="1024x1024",
         )
 
-    generate_button = gr.Button("Generate Image")
+    generate_button = gr.Button("Generate Image", variant="primary")
 
     output_image = gr.Image(
         label="Generated Image",
-        type="filepath",
     )
 
     status = gr.Textbox(
@@ -122,9 +160,21 @@ with gr.Blocks(title="SOLRICKS Image Generator") as demo:
 
     gr.Examples(
         examples=[
-            ["cinematic portrait of a futuristic woman, dramatic lighting, realistic details", "Portrait", "768x1344"],
-            ["luxury product photo of a perfume bottle, black background, studio lighting", "Product", "1024x1024"],
-            ["realistic AI robot in a dark premium technology lab, cinematic lighting", "Cinematic", "1344x768"],
+            [
+                "cinematic portrait of a futuristic woman, dramatic lighting, realistic details",
+                "Portrait",
+                "768x1344",
+            ],
+            [
+                "luxury product photo of a perfume bottle, black background, studio lighting",
+                "Product",
+                "1024x1024",
+            ],
+            [
+                "realistic AI robot in a dark premium technology lab, cinematic lighting",
+                "Cinematic",
+                "1344x768",
+            ],
         ],
         inputs=[prompt, style, resolution],
     )