Update app.py

app.py

@@ -6,10 +6,8 @@ import spaces
 
 from PIL import Image
 from diffusers import FlowMatchEulerDiscreteScheduler
-# from optimization import optimize_pipeline_
 from qwenimage.pipeline_qwenimage_edit_plus import QwenImageEditPlusPipeline
 from qwenimage.transformer_qwenimage import QwenImageTransformer2DModel
-# from qwenimage.qwen_fa3_processor import QwenDoubleStreamAttnProcessorFA3
 
 import math
 from huggingface_hub import hf_hub_download
@@ -50,15 +48,6 @@ pipe.unload_lora_weights()
 
 spaces.aoti_blocks_load(pipe.transformer, "zerogpu-aoti/Qwen-Image", variant="fa3")
 
-#pipe.transformer.__class__ = QwenImageTransformer2DModel
-#pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
-
-#optimize_pipeline_(
-#    pipe,
-#    image=[Image.new("RGB", (1024, 1024)), Image.new("RGB", (1024, 1024))],
-#    prompt="prompt"
-#)
-
 MAX_SEED = np.iinfo(np.int32).max
 
 
@@ -71,25 +60,6 @@ def _generate_video_segment(
     """
     Generate a single video segment between two frames by calling an external
     Wan 2.2 image-to-video service hosted on Hugging Face Spaces.
-
-    This helper function is used internally when the user asks to create
-    a video between the input and output images.
-
-    Args:
-        input_image_path (str):
-            Path to the starting frame image on disk.
-        output_image_path (str):
-            Path to the ending frame image on disk.
-        prompt (str):
-            Text prompt describing the camera movement / transition.
-        request (gr.Request):
-            Gradio request object, used here to forward the `x-ip-token`
-            header to the downstream Space for authentication/rate limiting.
-
-    Returns:
-        str:
-            A string returned by the external service, usually a URL or path
-            to the generated video.
     """
     x_ip_token = request.headers['x-ip-token']
     video_client = Client(
@@ -113,28 +83,6 @@ def build_camera_prompt(
 ) -> str:
     """
     Build a camera movement prompt based on the chosen controls.
-
-    This converts the provided control values into a prompt instruction with the corresponding trigger words for the multiple-angles LoRA.
-
-    Args:
-        rotate_deg (float, optional):
-            Horizontal rotation in degrees. Positive values rotate left,
-            negative values rotate right. Defaults to 0.0.
-        move_forward (float, optional):
-            Forward movement / zoom factor. Larger values imply moving the
-            camera closer or into a close-up. Defaults to 0.0.
-        vertical_tilt (float, optional):
-            Vertical angle of the camera:
-            - Negative ≈ bird's-eye view
-            - Positive ≈ worm's-eye view
-            Defaults to 0.0.
-        wideangle (bool, optional):
-            Whether to switch to a wide-angle lens style. Defaults to False.
-
-    Returns:
-        str:
-            A text prompt describing the camera motion. If no controls are
-            active, returns `"no camera movement"`.
     """
     prompt_parts = []
 
@@ -186,53 +134,7 @@ def infer_camera_edit(
     prev_output: Optional[Image.Image] = None,
 ) -> Tuple[Image.Image, int, str]:
     """
-    Edit the camera angles/view of an image with Qwen Image Edit 2509 and dx8152's Qwen-Edit-2509-Multiple-angles LoRA.
-
-    Applies a camera-style transformation (rotation, zoom, tilt, lens)
-    to an input image.
-
-    Args:
-        image (PIL.Image.Image | None, optional):
-            Input image to edit. If `None`, the function will instead try to
-            use `prev_output`. At least one of `image` or `prev_output` must
-            be available. Defaults to None.
-        rotate_deg (float, optional):
-            Horizontal rotation in degrees (-90, -45, 0, 45, 90). Positive values rotate
-            to the left, negative to the right. Defaults to 0.0.
-        move_forward (float, optional):
-            Forward movement / zoom factor (0, 5, 10). Higher values move the
-            camera closer; values >5 switch to a close-up style. Defaults to 0.0.
-        vertical_tilt (float, optional):
-            Vertical tilt (-1 to 1). -1 ≈ bird's-eye view, +1 ≈ worm's-eye view.
-            Defaults to 0.0.
-        wideangle (bool, optional):
-            Whether to use a wide-angle lens style. Defaults to False.
-        seed (int, optional):
-            Random seed for the generation. Ignored if `randomize_seed=True`.
-            Defaults to 0.
-        randomize_seed (bool, optional):
-            If True, a random seed (0..MAX_SEED) is chosen per call.
-            Defaults to True.
-        true_guidance_scale (float, optional):
-            CFG / guidance scale controlling prompt adherence.
-            Defaults to 1.0 since the demo is using a distilled transformer for faster inference.
-        num_inference_steps (int, optional):
-            Number of inference steps. Defaults to 4.
-        height (int, optional):
-            Output image height. Must typically be a multiple of 8.
-            If set to 0, the model will infer a size. Defaults to 1024 if none is provided.
-        width (int, optional):
-            Output image width. Must typically be a multiple of 8.
-            If set to 0, the model will infer a size. Defaults to 1024 if none is provided.
-        prev_output (PIL.Image.Image | None, optional):
-            Previous output image to use as input when no new image is uploaded.
-            Defaults to None.
-
-    Returns:
-        Tuple[PIL.Image.Image, int, str]:
-            - The edited output image.
-            - The actual seed used for generation.
-            - The constructed camera prompt string.
+    Edit the camera angles/view of an image with Qwen Image Edit 2509.
     """
     progress = gr.Progress(track_tqdm=True)
     
@@ -280,27 +182,7 @@ def create_video_between_images(
     request: gr.Request
 ) -> str:
     """
-    Create a short transition video between the input and output images via the 
-    Wan 2.2 first-last-frame Space.
-
-    Args:
-        input_image (PIL.Image.Image | None):
-            Starting frame image (the original / previous view).
-        output_image (numpy.ndarray | None):
-            Ending frame image - the output image with the the edited camera angles.
-        prompt (str):
-            The camera movement prompt used to describe the transition.
-        request (gr.Request):
-            Gradio request object, used to forward the `x-ip-token` header
-            to the video generation app.
-
-    Returns:
-        str:
-            a path pointing to the generated video.
-
-    Raises:
-        gr.Error:
-            If either image is missing or if the video generation fails.
+    Create a short transition video between the input and output images.
     """
     if input_image is None or output_image is None:
         raise gr.Error("Both input and output images are required to create a video.")
@@ -326,41 +208,504 @@ def create_video_between_images(
         raise gr.Error(f"Video generation failed: {e}")
 
 
-# --- UI ---
-css = '''#col-container { max-width: 800px; margin: 0 auto; }
-.dark .progress-text{color: white !important}
-#examples{max-width: 800px; margin: 0 auto; }'''
+# ============================================
+# 🎨 Comic Classic Theme - Toon Playground
+# ============================================
+
+css = """
+/* ===== 🎨 Google Fonts Import ===== */
+@import url('https://fonts.googleapis.com/css2?family=Bangers&family=Comic+Neue:wght@400;700&display=swap');
+
+/* ===== 🎨 Comic Classic 배경 - 빈티지 페이퍼 + 도트 패턴 ===== */
+.gradio-container {
+    background-color: #FEF9C3 !important;
+    background-image: 
+        radial-gradient(#1F2937 1px, transparent 1px) !important;
+    background-size: 20px 20px !important;
+    min-height: 100vh !important;
+    font-family: 'Comic Neue', cursive, sans-serif !important;
+}
+
+/* ===== 허깅페이스 상단 요소 숨김 ===== */
+.huggingface-space-header,
+#space-header,
+.space-header,
+[class*="space-header"],
+.svelte-1ed2p3z,
+.space-header-badge,
+.header-badge,
+[data-testid="space-header"],
+.svelte-kqij2n,
+.svelte-1ax1toq,
+.embed-container > div:first-child {
+    display: none !important;
+    visibility: hidden !important;
+    height: 0 !important;
+    width: 0 !important;
+    overflow: hidden !important;
+    opacity: 0 !important;
+    pointer-events: none !important;
+}
+
+/* ===== Footer 완전 숨김 ===== */
+footer,
+.footer,
+.gradio-container footer,
+.built-with,
+[class*="footer"],
+.gradio-footer,
+.main-footer,
+div[class*="footer"],
+.show-api,
+.built-with-gradio,
+a[href*="gradio.app"],
+a[href*="huggingface.co/spaces"] {
+    display: none !important;
+    visibility: hidden !important;
+    height: 0 !important;
+    padding: 0 !important;
+    margin: 0 !important;
+}
+
+/* ===== 메인 컨테이너 ===== */
+#col-container { 
+    max-width: 1000px; 
+    margin: 0 auto; 
+}
+
+/* ===== 🎨 헤더 타이틀 - 코믹 스타일 ===== */
+.header-text h1 {
+    font-family: 'Bangers', cursive !important;
+    color: #1F2937 !important;
+    font-size: 3.5rem !important;
+    font-weight: 400 !important;
+    text-align: center !important;
+    margin-bottom: 0.5rem !important;
+    text-shadow: 
+        4px 4px 0px #FACC15,
+        6px 6px 0px #1F2937 !important;
+    letter-spacing: 3px !important;
+    -webkit-text-stroke: 2px #1F2937 !important;
+}
+
+/* ===== 🎨 서브타이틀 ===== */
+.subtitle {
+    text-align: center !important;
+    font-family: 'Comic Neue', cursive !important;
+    font-size: 1.2rem !important;
+    color: #1F2937 !important;
+    margin-bottom: 1.5rem !important;
+    font-weight: 700 !important;
+}
+
+/* ===== 🎨 카드/패널 - 만화 프레임 스타일 ===== */
+.gr-panel,
+.gr-box,
+.gr-form,
+.block,
+.gr-group {
+    background: #FFFFFF !important;
+    border: 3px solid #1F2937 !important;
+    border-radius: 8px !important;
+    box-shadow: 6px 6px 0px #1F2937 !important;
+    transition: all 0.2s ease !important;
+}
+
+.gr-panel:hover,
+.block:hover {
+    transform: translate(-2px, -2px) !important;
+    box-shadow: 8px 8px 0px #1F2937 !important;
+}
+
+/* ===== 🎨 입력 필드 (Textbox) ===== */
+textarea, 
+input[type="text"], 
+input[type="number"] {
+    background: #FFFFFF !important;
+    border: 3px solid #1F2937 !important;
+    border-radius: 8px !important;
+    color: #1F2937 !important;
+    font-family: 'Comic Neue', cursive !important;
+    font-size: 1rem !important;
+    font-weight: 700 !important;
+    transition: all 0.2s ease !important;
+}
+
+textarea:focus, 
+input[type="text"]:focus, 
+input[type="number"]:focus {
+    border-color: #3B82F6 !important;
+    box-shadow: 4px 4px 0px #3B82F6 !important;
+    outline: none !important;
+}
+
+textarea::placeholder {
+    color: #9CA3AF !important;
+    font-weight: 400 !important;
+}
+
+/* ===== 🎨 Primary 버튼 - 코믹 블루 ===== */
+.gr-button-primary,
+button.primary,
+.gr-button.primary {
+    background: #3B82F6 !important;
+    border: 3px solid #1F2937 !important;
+    border-radius: 8px !important;
+    color: #FFFFFF !important;
+    font-family: 'Bangers', cursive !important;
+    font-weight: 400 !important;
+    font-size: 1.3rem !important;
+    letter-spacing: 2px !important;
+    padding: 14px 28px !important;
+    box-shadow: 5px 5px 0px #1F2937 !important;
+    transition: all 0.1s ease !important;
+    text-shadow: 1px 1px 0px #1F2937 !important;
+}
+
+.gr-button-primary:hover,
+button.primary:hover,
+.gr-button.primary:hover {
+    background: #2563EB !important;
+    transform: translate(-2px, -2px) !important;
+    box-shadow: 7px 7px 0px #1F2937 !important;
+}
+
+.gr-button-primary:active,
+button.primary:active,
+.gr-button.primary:active {
+    transform: translate(3px, 3px) !important;
+    box-shadow: 2px 2px 0px #1F2937 !important;
+}
+
+/* ===== 🎨 Secondary 버튼 - 코믹 레드 ===== */
+.gr-button-secondary,
+button.secondary,
+.generate-btn {
+    background: #EF4444 !important;
+    border: 3px solid #1F2937 !important;
+    border-radius: 8px !important;
+    color: #FFFFFF !important;
+    font-family: 'Bangers', cursive !important;
+    font-weight: 400 !important;
+    font-size: 1.1rem !important;
+    letter-spacing: 1px !important;
+    box-shadow: 4px 4px 0px #1F2937 !important;
+    transition: all 0.1s ease !important;
+    text-shadow: 1px 1px 0px #1F2937 !important;
+}
+
+.gr-button-secondary:hover,
+button.secondary:hover,
+.generate-btn:hover {
+    background: #DC2626 !important;
+    transform: translate(-2px, -2px) !important;
+    box-shadow: 6px 6px 0px #1F2937 !important;
+}
+
+.gr-button-secondary:active,
+button.secondary:active,
+.generate-btn:active {
+    transform: translate(2px, 2px) !important;
+    box-shadow: 2px 2px 0px #1F2937 !important;
+}
+
+/* ===== 🎨 Reset 버튼 - 코믹 옐로우 ===== */
+.reset-btn {
+    background: #FACC15 !important;
+    border: 3px solid #1F2937 !important;
+    border-radius: 8px !important;
+    color: #1F2937 !important;
+    font-family: 'Bangers', cursive !important;
+    font-weight: 400 !important;
+    font-size: 1.1rem !important;
+    letter-spacing: 1px !important;
+    box-shadow: 4px 4px 0px #1F2937 !important;
+    transition: all 0.1s ease !important;
+}
+
+.reset-btn:hover {
+    background: #EAB308 !important;
+    transform: translate(-2px, -2px) !important;
+    box-shadow: 6px 6px 0px #1F2937 !important;
+}
+
+.reset-btn:active {
+    transform: translate(2px, 2px) !important;
+    box-shadow: 2px 2px 0px #1F2937 !important;
+}
+
+/* ===== 🎨 Video 버튼 - 코믹 퍼플 ===== */
+.video-btn {
+    background: #8B5CF6 !important;
+    border: 3px solid #1F2937 !important;
+    border-radius: 8px !important;
+    color: #FFFFFF !important;
+    font-family: 'Bangers', cursive !important;
+    font-weight: 400 !important;
+    font-size: 1.1rem !important;
+    letter-spacing: 1px !important;
+    box-shadow: 4px 4px 0px #1F2937 !important;
+    transition: all 0.1s ease !important;
+    text-shadow: 1px 1px 0px #1F2937 !important;
+}
+
+.video-btn:hover {
+    background: #7C3AED !important;
+    transform: translate(-2px, -2px) !important;
+    box-shadow: 6px 6px 0px #1F2937 !important;
+}
+
+.video-btn:active {
+    transform: translate(2px, 2px) !important;
+    box-shadow: 2px 2px 0px #1F2937 !important;
+}
+
+/* ===== 🎨 로그 출력 영역 ===== */
+.info-log textarea,
+.prompt-preview textarea {
+    background: #1F2937 !important;
+    color: #10B981 !important;
+    font-family: 'Courier New', monospace !important;
+    font-size: 0.9rem !important;
+    font-weight: 400 !important;
+    border: 3px solid #10B981 !important;
+    border-radius: 8px !important;
+    box-shadow: 4px 4px 0px #10B981 !important;
+}
+
+/* ===== 🎨 이미지 업로드 영역 ===== */
+.image-upload {
+    border: 4px dashed #3B82F6 !important;
+    border-radius: 12px !important;
+    background: #EFF6FF !important;
+    transition: all 0.2s ease !important;
+}
+
+.image-upload:hover {
+    border-color: #EF4444 !important;
+    background: #FEF2F2 !important;
+}
+
+/* ===== 🎨 아코디언 - 말풍선 스타일 ===== */
+.gr-accordion {
+    background: #FACC15 !important;
+    border: 3px solid #1F2937 !important;
+    border-radius: 8px !important;
+    box-shadow: 4px 4px 0px #1F2937 !important;
+}
+
+.gr-accordion-header {
+    color: #1F2937 !important;
+    font-family: 'Comic Neue', cursive !important;
+    font-weight: 700 !important;
+    font-size: 1.1rem !important;
+}
+
+/* ===== 🎨 탭 스타일 - 코믹 스타일 ===== */
+.gr-tab-nav {
+    background: #FFFFFF !important;
+    border: 3px solid #1F2937 !important;
+    border-radius: 8px 8px 0 0 !important;
+    box-shadow: 4px 0px 0px #1F2937 !important;
+}
+
+.gr-tab-nav button {
+    font-family: 'Comic Neue', cursive !important;
+    font-weight: 700 !important;
+    color: #1F2937 !important;
+    border: none !important;
+    background: transparent !important;
+}
+
+.gr-tab-nav button.selected {
+    background: #3B82F6 !important;
+    color: #FFFFFF !important;
+    border-radius: 6px !important;
+}
+
+/* ===== 🎨 슬라이더 스타일 ===== */
+input[type="range"] {
+    accent-color: #3B82F6 !important;
+}
+
+.gr-slider {
+    background: #FFFFFF !important;
+}
+
+/* ===== 🎨 체크박스 스타일 ===== */
+input[type="checkbox"] {
+    accent-color: #3B82F6 !important;
+    width: 20px !important;
+    height: 20px !important;
+    border: 2px solid #1F2937 !important;
+    border-radius: 4px !important;
+}
+
+/* ===== 🎨 이미지 출력 영역 ===== */
+.gr-image,
+.image-container {
+    border: 4px solid #1F2937 !important;
+    border-radius: 8px !important;
+    box-shadow: 8px 8px 0px #1F2937 !important;
+    overflow: hidden !important;
+    background: #FFFFFF !important;
+}
+
+/* ===== 🎨 비디오 출력 영역 ===== */
+.gr-video {
+    border: 4px solid #1F2937 !important;
+    border-radius: 8px !important;
+    box-shadow: 8px 8px 0px #8B5CF6 !important;
+    overflow: hidden !important;
+    background: #FFFFFF !important;
+}
+
+/* ===== 🎨 라벨 스타일 ===== */
+label,
+.gr-input-label,
+.gr-block-label {
+    color: #1F2937 !important;
+    font-family: 'Comic Neue', cursive !important;
+    font-weight: 700 !important;
+    font-size: 1rem !important;
+}
+
+span.gr-label {
+    color: #1F2937 !important;
+}
+
+/* ===== 🎨 정보 텍스트 ===== */
+.gr-info,
+.info {
+    color: #6B7280 !important;
+    font-family: 'Comic Neue', cursive !important;
+    font-size: 0.9rem !important;
+}
+
+/* ===== 🎨 프로그레스 바 ===== */
+.progress-bar,
+.gr-progress-bar {
+    background: #3B82F6 !important;
+    border: 2px solid #1F2937 !important;
+    border-radius: 4px !important;
+}
+
+.progress-text {
+    color: #1F2937 !important;
+    font-family: 'Comic Neue', cursive !important;
+    font-weight: 700 !important;
+}
+
+/* ===== 🎨 스크롤바 - 코믹 스타일 ===== */
+::-webkit-scrollbar {
+    width: 12px;
+    height: 12px;
+}
+
+::-webkit-scrollbar-track {
+    background: #FEF9C3;
+    border: 2px solid #1F2937;
+}
+
+::-webkit-scrollbar-thumb {
+    background: #3B82F6;
+    border: 2px solid #1F2937;
+    border-radius: 0px;
+}
+
+::-webkit-scrollbar-thumb:hover {
+    background: #EF4444;
+}
+
+/* ===== 🎨 선택 하이라이트 ===== */
+::selection {
+    background: #FACC15;
+    color: #1F2937;
+}
+
+/* ===== 🎨 링크 스타일 ===== */
+a {
+    color: #3B82F6 !important;
+    text-decoration: none !important;
+    font-weight: 700 !important;
+}
+
+a:hover {
+    color: #EF4444 !important;
+}
+
+/* ===== 🎨 Row/Column 간격 ===== */
+.gr-row {
+    gap: 1.5rem !important;
+}
+
+.gr-column {
+    gap: 1rem !important;
+}
+
+/* ===== 🎨 Examples 섹션 ===== */
+#examples {
+    max-width: 1000px;
+    margin: 0 auto;
+}
+
+#examples .gr-sample {
+    border: 3px solid #1F2937 !important;
+    border-radius: 8px !important;
+    box-shadow: 4px 4px 0px #1F2937 !important;
+    background: #FFFFFF !important;
+    transition: all 0.2s ease !important;
+}
+
+#examples .gr-sample:hover {
+    transform: translate(-2px, -2px) !important;
+    box-shadow: 6px 6px 0px #1F2937 !important;
+}
+
+/* ===== 반응형 조정 ===== */
+@media (max-width: 768px) {
+    .header-text h1 {
+        font-size: 2.2rem !important;
+        text-shadow: 
+            3px 3px 0px #FACC15,
+            4px 4px 0px #1F2937 !important;
+    }
+    
+    .gr-button-primary,
+    button.primary {
+        padding: 12px 20px !important;
+        font-size: 1.1rem !important;
+    }
+    
+    .gr-panel,
+    .block {
+        box-shadow: 4px 4px 0px #1F2937 !important;
+    }
+}
+
+/* ===== 🎨 다크모드 비활성화 (코믹은 밝아야 함) ===== */
+@media (prefers-color-scheme: dark) {
+    .gradio-container {
+        background-color: #FEF9C3 !important;
+    }
+    
+    .progress-text {
+        color: #1F2937 !important;
+    }
+}
+"""
 
 
 def reset_all() -> list:
     """
     Reset all camera control knobs and flags to their default values.
-
-    This is used by the "Reset" button to set:
-    - rotate_deg = 0
-    - move_forward = 0
-    - vertical_tilt = 0
-    - wideangle = False
-    - is_reset = True
-
-    Returns:
-        list:
-            A list of values matching the order of the reset outputs:
-            [rotate_deg, move_forward, vertical_tilt, wideangle, is_reset, True]
     """
-    return [0, 0, 0, 0, False, True]
+    return [0, 0, 0, False, True]
 
 
 def end_reset() -> bool:
     """
     Mark the end of a reset cycle.
-
-    This helper is chained after `reset_all` to set the internal
-    `is_reset` flag back to False, so that live inference can resume.
-
-    Returns:
-        bool:
-            Always returns False.
     """
     return False
 
@@ -369,16 +714,7 @@ def update_dimensions_on_upload(
     image: Optional[Image.Image]
 ) -> Tuple[int, int]:
     """
-    Compute recommended (width, height) for the output resolution when an
-    image is uploaded while preserveing the aspect ratio.
-
-    Args:
-        image (PIL.Image.Image | None):
-            The uploaded image. If `None`, defaults to (1024, 1024).
-
-    Returns:
-        Tuple[int, int]:
-            The new (width, height).
+    Compute recommended (width, height) for the output resolution.
     """
     if image is None:
         return 1024, 1024
@@ -401,99 +737,160 @@ def update_dimensions_on_upload(
     return new_width, new_height
 
 
-with gr.Blocks() as demo:
+# Build the Gradio interface with Comic Classic Theme
+with gr.Blocks(fill_height=True, css=css) as demo:
+    
+    # HOME Badge
+    gr.HTML("""
+        <div style="text-align: center; margin: 20px 0 10px 0;">
+            <a href="https://www.humangen.ai" target="_blank" style="text-decoration: none;">
+                <img src="https://img.shields.io/static/v1?label=🏠 HOME&message=HUMANGEN.AI&color=0000ff&labelColor=ffcc00&style=for-the-badge" alt="HOME">
+            </a>
+        </div>
+    """)
+    
+    # Header Title
+    gr.Markdown(
+        """
+        # 🎬 CAMERA ANGLE CONTROL 📷
+        """,
+        elem_classes="header-text"
+    )
+    
+    gr.Markdown(
+        """
+        <p class="subtitle">🖼️ Upload an image and control camera angles with AI magic! ✨</p>
+        """,
+    )
+    
     with gr.Column(elem_id="col-container"):
-        gr.Markdown("## 🎬 Qwen Image Edit — Camera Angle Control")
-
-
-        with gr.Row():
-            with gr.Column():
-                image = gr.Image(label="Input Image", type="pil")
+        with gr.Row(equal_height=False):
+            # Left column - Input
+            with gr.Column(scale=1, min_width=320):
+                image = gr.Image(
+                    label="📷 Upload Your Image",
+                    type="pil",
+                    elem_classes="image-upload"
+                )
                 prev_output = gr.Image(value=None, visible=False)
                 is_reset = gr.Checkbox(value=False, visible=False)
-
-                with gr.Tab("Camera Controls"):
+                
+                with gr.Accordion("🎮 Camera Controls", open=True):
                     rotate_deg = gr.Slider(
-                        label="Rotate Right-Left (degrees °)",
+                        label="↔️ Rotate Right-Left (degrees °)",
                         minimum=-90,
                         maximum=90,
                         step=45,
                         value=0
                     )
                     move_forward = gr.Slider(
-                        label="Move Forward → Close-Up",
+                        label="🔍 Move Forward → Close-Up",
                         minimum=0,
                         maximum=10,
                         step=5,
                         value=0
                     )
                     vertical_tilt = gr.Slider(
-                        label="Vertical Angle (Bird ↔ Worm)",
+                        label="↕️ Vertical Angle (Bird ↔ Worm)",
                         minimum=-1,
                         maximum=1,
                         step=1,
                         value=0
                     )
-                    wideangle = gr.Checkbox(label="Wide-Angle Lens", value=False)
+                    wideangle = gr.Checkbox(
+                        label="📐 Wide-Angle Lens",
+                        value=False
+                    )
+                
                 with gr.Row():
-                    reset_btn = gr.Button("Reset")
-                    run_btn = gr.Button("Generate", variant="primary")
-
-                with gr.Accordion("Advanced Settings", open=False):
+                    reset_btn = gr.Button(
+                        "🔄 RESET",
+                        variant="secondary",
+                        elem_classes="reset-btn"
+                    )
+                    run_btn = gr.Button(
+                        "🎬 GENERATE!",
+                        variant="primary",
+                        size="lg",
+                        elem_classes="generate-btn"
+                    )
+                
+                with gr.Accordion("⚙️ Advanced Settings", open=False):
                     seed = gr.Slider(
-                        label="Seed",
+                        label="🎲 Seed",
                         minimum=0,
                         maximum=MAX_SEED,
                         step=1,
                         value=0
                     )
                     randomize_seed = gr.Checkbox(
-                        label="Randomize Seed",
+                        label="🔀 Randomize Seed",
                         value=True
                     )
                     true_guidance_scale = gr.Slider(
-                        label="True Guidance Scale",
+                        label="📊 True Guidance Scale",
                         minimum=1.0,
                         maximum=10.0,
                         step=0.1,
                         value=1.0
                     )
                     num_inference_steps = gr.Slider(
-                        label="Inference Steps",
+                        label="🔢 Inference Steps",
                         minimum=1,
                         maximum=40,
                         step=1,
                         value=20
                     )
                     height = gr.Slider(
-                        label="Height",
+                        label="📏 Height",
                         minimum=256,
                         maximum=2048,
                         step=8,
                         value=1024
                     )
                     width = gr.Slider(
-                        label="Width",
+                        label="📐 Width",
                         minimum=256,
                         maximum=2048,
                         step=8,
                         value=1024
                     )
-
-            with gr.Column():
-                result = gr.Image(label="Output Image", interactive=False)
-                prompt_preview = gr.Textbox(label="Processed Prompt", interactive=False)
+            
+            # Right column - Output
+            with gr.Column(scale=1, min_width=320):
+                result = gr.Image(
+                    label="🖼️ Output Image",
+                    interactive=False,
+                    height=400,
+                )
+                
+                prompt_preview = gr.Textbox(
+                    label="📝 Generated Prompt",
+                    interactive=False,
+                    lines=3,
+                    elem_classes="prompt-preview"
+                )
+                
                 create_video_button = gr.Button(
-                    "🎥 Create Video Between Images",
+                    "🎥 CREATE VIDEO BETWEEN IMAGES!",
                     variant="secondary",
-                    visible=False
+                    visible=False,
+                    elem_classes="video-btn"
                 )
+                
                 with gr.Group(visible=False) as video_group:
                     video_output = gr.Video(
-                        label="Generated Video",
-                        buttons=["download"],
+                        label="🎬 Generated Video",
                         autoplay=True
                     )
+                
+                gr.Markdown(
+                    """
+                    <p style="text-align: center; margin-top: 10px; font-weight: 700; color: #1F2937;">
+                        💡 Right-click on the image to save, or use the download button!
+                    </p>
+                    """
+                )
 
     inputs = [
         image, rotate_deg, move_forward,
@@ -513,22 +910,10 @@ with gr.Blocks() as demo:
     # Manual generation with video button visibility control
     def infer_and_show_video_button(*args: Any):
         """
-        Wrapper around `infer_camera_edit` that also controls the visibility
+        Wrapper around infer_camera_edit that also controls the visibility
         of the 'Create Video Between Images' button.
-
-        The first argument in `args` is expected to be the input image; if both
-        input and output images are present, the video button is shown.
-
-        Args:
-            *args:
-                Positional arguments forwarded directly to `infer_camera_edit`.
-
-        Returns:
-            tuple:
-                (output_image, seed, prompt, video_button_visibility_update)
         """
         result_img, result_seed, result_prompt = infer_camera_edit(*args)
-        # Show video button if we have both input and output images
         show_button = args[0] is not None and result_img is not None
         return result_img, result_seed, result_prompt, gr.update(visible=show_button)
 
@@ -598,7 +983,6 @@ with gr.Blocks() as demo:
             return gr.update(), gr.update(), gr.update(), gr.update()
         else:
             result_img, result_seed, result_prompt = infer_camera_edit(*args)
-            # Show video button if we have both input and output
             show_button = args[0] is not None and result_img is not None
             return result_img, result_seed, result_prompt, gr.update(visible=show_button)
 
@@ -627,4 +1011,5 @@ with gr.Blocks() as demo:
     gr.api(infer_camera_edit, api_name="infer_edit_camera_angles")
     gr.api(create_video_between_images, api_name="create_video_between_images")
 
-demo.launch(mcp_server=True, theme=gr.themes.Citrus(), css=css, footer_links=["api", "gradio", "settings"])
+if __name__ == "__main__":
+    demo.launch(mcp_server=True)