Update app.py

app.py

@@ -1,142 +1,62 @@
 import gradio as gr
-import numpy as np
-import random
-from PIL import Image
-from diffusers import StableDiffusionXLPipeline
-import torch
-import gc
+import os
 
-# System configuration
-device = "cpu"
-model_repo_id = "stabilityai/sdxl-turbo"
+URL_EDITOR = "https://selfit-camera-omni-image-editor.hf.space/?__theme=dark"
+URL_VIDEO = "https://frameai4687-omni-video-factory.hf.space/?__theme=dark"
 
-# 1. Load SDXL Turbo Model
-pipe = StableDiffusionXLPipeline.from_pretrained(
-    model_repo_id,
-    torch_dtype=torch.float32,
-    use_safetensors=True
-)
-pipe.enable_attention_slicing()
-
-MAX_SEED = np.iinfo(np.int32).max
-
-# Style templates
-STYLE_TEMPLATES = {
-    "None": "{prompt}",
-    "Cinematic": "{prompt}, cinematic lighting, highly detailed, 8k, masterpiece",
-    "Anime": "{prompt}, anime style, vibrant colors, clean lines, high quality",
-    "Digital Art": "{prompt}, digital illustration, sharp focus, trend on artstation",
-    "Oil Painting": "{prompt}, oil painting texture, brush strokes, classical art",
-    "Photorealistic": "{prompt}, ultra realistic, photorealistic, raw photo, 8k uhd"
+css = """
+.gradio-container { 
+    max-width: 100% !important; 
+    padding: 0 !important; 
+    background-color: #0b0f19 !important; 
+}
+#status-bar {
+    background: linear-gradient(90deg, #1e293b 0%, #0f172a 100%);
+    color: #38bdf8;
+    padding: 10px;
+    font-weight: bold;
+    border-bottom: 2px solid #38bdf8;
+    text-align: center;
+    font-size: 1em;
+    /* รองรับพื้นที่ปลอดภัยสำหรับมือถือที่มีติ่งหน้าจอ */
+    padding-top: env(safe-area-inset-top, 10px); 
+}
+iframe {
+    width: 100%;
+    height: 85vh;
+    border: none;
+}
+/* ปรับแต่งสำหรับหน้าจอขนาดเล็ก (Mobile) */
+@media (max-width: 768px) {
+    #status-bar { font-size: 0.8em; }
+    iframe { height: 80vh; }
+    .tabs button { font-size: 0.7em !important; padding: 5px !important; }
 }
-
-def update_resolution(aspect_ratio):
-    if aspect_ratio == "1:1 (512x512)":
-        return 512, 512
-    elif aspect_ratio == "4:3 (512x384)":
-        return 512, 384
-    elif aspect_ratio == "3:4 (384x512)":
-        return 384, 512
-    elif aspect_ratio == "16:9 (512x288)":
-        return 512, 288
-    return 512, 512
-
-def infer(
-    prompt,
-    negative_prompt,
-    seed,
-    randomize_seed,
-    width,
-    height,
-    guidance_scale,
-    num_inference_steps,
-    style_selection,
-    auto_enhance
-):
-    if randomize_seed:
-        seed = random.randint(0, MAX_SEED)
-    
-    generator = torch.Generator(device=device).manual_seed(seed)
-    
-    # Apply style and auto-enhancement
-    final_prompt = STYLE_TEMPLATES.get(style_selection, "{prompt}").format(prompt=prompt)
-    if auto_enhance:
-        final_prompt += ", professional lighting, high resolution, intricate details"
-
-    try:
-        image = pipe(
-            prompt=final_prompt,
-            negative_prompt=negative_prompt,
-            guidance_scale=guidance_scale,
-            num_inference_steps=num_inference_steps,
-            width=width,
-            height=height,
-            generator=generator,
-        ).images[0]
-    except Exception as e:
-        print(f"Inference Error: {e}")
-        return None, seed
-    finally:
-        gc.collect()
-            
-    return image, seed
-
-# UI Styling
-custom_css = """
-body, .gradio-container { background-color: #0b0b0b !important; color: #ff0000 !important; }
-button.primary { background: linear-gradient(90deg, #8b0000, #ff0000) !important; color: white !important; border: none !important; }
-footer { visibility: hidden; }
-.tabs, .accordion, .input-container { border: 1px solid #440000 !important; }
 """
 
-with gr.Blocks(css=custom_css, theme=gr.themes.Soft(primary_hue="red", neutral_hue="slate")) as demo:
-    gr.Markdown("# <span style='color: #ff0000;'>Z-TURBO RED ENGINE</span>")
-    
-    with gr.Row():
-        with gr.Column(scale=3):
-            prompt = gr.Textbox(label="Main Prompt", placeholder="What do you want to create?", lines=3)
-            style_selection = gr.Dropdown(choices=list(STYLE_TEMPLATES.keys()), value="None", label="Art Style")
-            auto_enhance = gr.Checkbox(label="Auto Prompt Enhancement", value=False)
-            run_button = gr.Button("GENERATE IMAGE", variant="primary")
-            
-        with gr.Column(scale=2):
-            result = gr.Image(label="Result", type="pil")
-            seed_display = gr.Number(label="Last Seed Used", interactive=False)
+js_func = """
+function() {
+    const tabs = document.querySelectorAll('.tabs button');
+    const statusBar = document.querySelector('#status-bar span');
+    if(tabs && statusBar) {
+        tabs.forEach(tab => {
+            tab.addEventListener('click', () => {
+                statusBar.innerText = tab.innerText.toUpperCase();
+            });
+        });
+    }
+}
+"""
 
-    with gr.Row():
-        with gr.Accordion("Advanced Parameters", open=True):
-            with gr.Row():
-                aspect_ratio = gr.Radio(
-                    choices=["1:1 (512x512)", "4:3 (512x384)", "3:4 (384x512)", "16:9 (512x288)"],
-                    value="1:1 (512x512)",
-                    label="Quick Aspect Ratio"
-                )
-            with gr.Row():
-                width = gr.Slider(label="Width", minimum=256, maximum=512, step=64, value=512)
-                height = gr.Slider(label="Height", minimum=256, maximum=512, step=64, value=512)
-            
-            with gr.Row():
-                guidance_scale = gr.Slider(label="Guidance Scale", minimum=0.0, maximum=5.0, step=0.1, value=1.0)
-                num_inference_steps = gr.Slider(label="Steps", minimum=1, maximum=20, step=1, value=8)
-                
-            with gr.Row():
-                seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0)
-                randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
+with gr.Blocks(css=css, title="Omni Studio Pro", js=js_func) as demo:
+    gr.HTML('<div id="status-bar">SYSTEM ACTIVE: <span>🎨 OMNI IMAGE EDITOR</span></div>')
 
-    # Negative Prompt at the very bottom
-    negative_prompt = gr.Textbox(label="Negative Prompt (Short)", value="blurry, ugly, distorted", lines=1)
+    with gr.Tabs() as tabs:
+        with gr.Tab("🎨 Omni Image Editor"):
+            gr.HTML(f'<iframe src="{URL_EDITOR}" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share" allowfullscreen></iframe>')
 
-    # Event handlers
-    aspect_ratio.change(fn=update_resolution, inputs=[aspect_ratio], outputs=[width, height])
-    
-    run_button.click(
-        fn=infer,
-        inputs=[
-            prompt, negative_prompt, seed, randomize_seed, 
-            width, height, guidance_scale, num_inference_steps,
-            style_selection, auto_enhance
-        ],
-        outputs=[result, seed_display],
-    )
+        with gr.Tab("🎬 Omni Video Factory"):
+            gr.HTML(f'<iframe src="{URL_VIDEO}" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share" allowfullscreen></iframe>')
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", "7860")))