Improve UI design with modern styling, better layout, and visual enhancements

app.py

@@ -292,14 +292,93 @@ def generate_video(prompt, aspect_ratio, cfg, seed, target_length, checkpoint_fi
         return None, f"Error: {str(e)}"
 
 
-# Create Gradio interface
-with gr.Blocks(title="STARFlow - Text-to-Image & Video Generation") as demo:
-    gr.Markdown("""
-    # STARFlow: Scalable Transformer Auto-Regressive Flow
-    
-    Generate high-quality images and videos from text prompts using STARFlow models.
+# Create Gradio interface with custom CSS
+custom_css = """
+    .gradio-container {
+        font-family: 'Inter', -apple-system, BlinkMacSystemFont, 'Segoe UI', sans-serif;
+    }
+    .main-header {
+        text-align: center;
+        padding: 2rem 0;
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+        color: white;
+        border-radius: 10px;
+        margin-bottom: 2rem;
+    }
+    .main-header h1 {
+        margin: 0;
+        font-size: 2.5rem;
+        font-weight: 700;
+        text-shadow: 2px 2px 4px rgba(0,0,0,0.2);
+    }
+    .main-header p {
+        margin: 0.5rem 0 0 0;
+        font-size: 1.1rem;
+        opacity: 0.95;
+    }
+    .info-box {
+        background: #f0f4ff;
+        border-left: 4px solid #667eea;
+        padding: 1rem;
+        border-radius: 5px;
+        margin-bottom: 1.5rem;
+    }
+    .input-section {
+        background: #fafafa;
+        padding: 1.5rem;
+        border-radius: 10px;
+        border: 1px solid #e0e0e0;
+    }
+    .output-section {
+        background: white;
+        padding: 1.5rem;
+        border-radius: 10px;
+        border: 1px solid #e0e0e0;
+        box-shadow: 0 2px 8px rgba(0,0,0,0.05);
+    }
+    .generate-btn {
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
+        color: white !important;
+        font-weight: 600 !important;
+        padding: 0.75rem 2rem !important;
+        border-radius: 8px !important;
+        border: none !important;
+        font-size: 1rem !important;
+        transition: transform 0.2s, box-shadow 0.2s !important;
+    }
+    .generate-btn:hover {
+        transform: translateY(-2px);
+        box-shadow: 0 4px 12px rgba(102, 126, 234, 0.4) !important;
+    }
+    .status-box {
+        font-family: 'Monaco', 'Menlo', monospace;
+        font-size: 0.9rem;
+        line-height: 1.6;
+    }
+"""
+
+with gr.Blocks(
+    title="STARFlow - Text-to-Image & Video Generation",
+    theme=gr.themes.Soft(primary_hue="purple"),
+    css=custom_css
+) as demo:
+    # Header
+    gr.HTML("""
+    <div class="main-header">
+        <h1>🎨 STARFlow</h1>
+        <p>Scalable Transformer Auto-Regressive Flow</p>
+        <p style="font-size: 0.95rem; margin-top: 0.5rem; opacity: 0.9;">
+            Generate high-quality images and videos from text prompts
+        </p>
+    </div>
+    """)
     
-    **Checkpoints are automatically downloaded from Model Hub on first use.**
+    # Info box
+    gr.Markdown("""
+    <div class="info-box">
+        <strong>ℹ️ Note:</strong> Checkpoints are automatically downloaded from Model Hub on first use. 
+        First generation may take 10-20 minutes for download and model loading.
+    </div>
     """)
     
     with gr.Tabs():