Upload 2 files

app.py

@@ -0,0 +1,76 @@
+import gradio as gr
+import spaces
+from huggingface_hub import hf_hub_download
+import os
+
+# ====================== DOWNLOAD MODELOV ======================
+def download_models():
+    print("📥 Downloading required models... (first run may take time)")
+    
+    # Qwen Image Edit models
+    hf_hub_download("Comfy-Org/Qwen-Image-Edit_ComfyUI", 
+                   "split_files/diffusion_models/qwen_image_edit_2509_fp8_e4m3fn.safetensors", 
+                   local_dir="models/diffusion_models", force_download=False)
+    
+    hf_hub_download("Comfy-Org/Qwen-Image_ComfyUI", 
+                   "split_files/vae/qwen_image_vae.safetensors", 
+                   local_dir="models/vae", force_download=False)
+    
+    hf_hub_download("Comfy-Org/Qwen-Image_ComfyUI", 
+                   "split_files/text_encoders/qwen_2.5_vl_7b_fp8_scaled.safetensors", 
+                   local_dir="models/text_encoders", force_download=False)
+    
+    hf_hub_download("lightx2v/Qwen-Image-Lightning", 
+                   "Qwen-Image-Edit-2509/Qwen-Image-Edit-2509-Lightning-4steps-V1.0-bf16.safetensors", 
+                   local_dir="models/loras", force_download=False)
+
+    print("✅ Models ready!")
+
+download_models()
+
+# ====================== GENERATE FUNCTION ======================
+@spaces.GPU(duration=180)
+def generate(prompt_image, prompt_video, bedroom_image, woman_image):
+    if bedroom_image is None or woman_image is None:
+        return None, None, "❌ Naloži obe sliki!"
+    
+    status = "🔄 Processing...\n1. Qwen Image Edit\n2. Wan2.2 Image to Video"
+    # Tukaj bo kasneje prišla polna ComfyUI logika
+    # Za zdaj placeholder
+    
+    return None, None, status + "\n\n(Full execution coming soon)"
+
+# ====================== GRADIO UI ======================
+with gr.Blocks(title="Bedroom → Video", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🛏️ Bedroom Mix + Wan2.2 Video Generator")
+    
+    with gr.Row():
+        with gr.Column():
+            prompt_image = gr.Textbox(
+                label="Prompt za urejanje slike (Qwen)",
+                lines=4,
+                value="Take the bedroom from image 1 and naturally integrate the woman from image 2 on the bed. Realistic proportions, good lighting, intimate atmosphere."
+            )
+            prompt_video = gr.Textbox(
+                label="Prompt za video (Wan2.2)",
+                lines=4,
+                value="smooth cinematic motion, realistic physics, intimate scene, high detail, 8k"
+            )
+            
+            with gr.Row():
+                img1 = gr.Image(label="Bedroom Image", type="filepath")
+                img2 = gr.Image(label="Woman Image", type="filepath")
+            
+            btn = gr.Button("🚀 Generate", variant="primary", size="large")
+        
+        with gr.Column():
+            out_img = gr.Image(label="Generated Image")
+            out_video = gr.Video(label="Generated Video")
+            status = gr.Textbox(label="Status", interactive=False)
+
+    btn.click(generate, 
+              inputs=[prompt_image, prompt_video, img1, img2],
+              outputs=[out_img, out_video, status])
+
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)

requirements.txt

@@ -0,0 +1,11 @@
+gradio>=5.0.0
+spaces>=0.30.0
+huggingface_hub>=0.25.0
+Pillow
+numpy
+torch
+torchvision
+torchaudio
+opencv-python-headless
+imageio
+imageio-ffmpeg