add separate boxes for prompts

app.py

@@ -22,7 +22,7 @@ luma_client = LumaAI(auth_token=LUMA_API_KEY)
 # PHASE 1: GPU DRAWING (ZeroGPU)
 # ==========================================
 @spaces.GPU(duration=60)
-def generate_frames(sketch_1, sketch_2, prompt, ctrl_scale, consistency):
+def generate_frames(sketch_1, sketch_2, image_prompt, ctrl_scale, consistency):
     print("🎨 Initializing GPU Pipelines...")
     dtype = torch.float16
     controlnet = ControlNetModel.from_pretrained("xinsir/controlnet-scribble-sdxl-1.0", torch_dtype=dtype)
@@ -37,7 +37,7 @@ def generate_frames(sketch_1, sketch_2, prompt, ctrl_scale, consistency):
     # --- Start Frame ---
     s1 = sketch_1.convert("RGB").resize((1024, 1024), Image.NEAREST)
     img_1 = pipe(
-        prompt=f"stunning digital illustration, {prompt}", 
+        prompt=f"stunning digital illustration, {image_prompt}", 
         image=s1, 
         controlnet_conditioning_scale=float(ctrl_scale)
     ).images[0]
@@ -47,7 +47,7 @@ def generate_frames(sketch_1, sketch_2, prompt, ctrl_scale, consistency):
     i2i_pipe = StableDiffusionXLControlNetImg2ImgPipeline(**pipe.components).to("cuda")
     s2 = sketch_2.convert("RGB").resize((1024, 1024), Image.NEAREST)
     img_2 = i2i_pipe(
-        prompt=f"stunning digital illustration, {prompt}", 
+        prompt=f"stunning digital illustration, {image_prompt}", 
         image=img_1, 
         control_image=s2, 
         strength=float(consistency)
@@ -68,12 +68,12 @@ def upload_to_web(image):
         )
     return response.json()['data']['url']
 
-def master_pipeline(sketch_1, sketch_2, prompt, ctrl_scale, consistency):
+def master_pipeline(sketch_1, sketch_2, image_prompt, video_prompt, ctrl_scale, consistency):
     if not LUMA_API_KEY or not IMGBB_API_KEY:
         raise gr.Error("🔑 API Keys missing! Add them to Space Secrets.")
-
+        
     # 1. Generate Images on ZeroGPU
-    img_1, img_2 = generate_frames(sketch_1, sketch_2, prompt, ctrl_scale, consistency)
+    img_1, img_2 = generate_frames(sketch_1, sketch_2, image_prompt, ctrl_scale, consistency)
     
     # 2. Upload to ImgBB (CPU)
     print("☁️ Uploading to cloud...")
@@ -83,7 +83,7 @@ def master_pipeline(sketch_1, sketch_2, prompt, ctrl_scale, consistency):
     # 3. Request Luma Video (CPU)
     print("🎬 Requesting Luma Ray-2 Animation...")
     generation = luma_client.generations.create(
-        prompt=f"Cinematic stylized animation, {prompt}",
+        prompt=f"Cinematic stylized animation, {video_prompt}",
         model="ray-2",
         duration="3s",
         resolution="540p",
@@ -110,21 +110,36 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("Transform sketches into professional animations using local ZeroGPU stylization and Luma cloud rendering.")
     
     with gr.Row():
+        # --- INPUT COLUMN ---
         with gr.Column():
             s1 = gr.Image(type="pil", label="Start Sketch (Black & White)")
             s2 = gr.Image(type="pil", label="End Sketch (Black & White)")
-            p = gr.Textbox(label="Prompt", placeholder="e.g., Plague doctor, bird mask, chanting")
+            
+            # Separated Prompts
+            img_prompt = gr.Textbox(label="Image Prompt", placeholder="e.g., Plague doctor, bird mask, cinematic lighting")
+            vid_prompt = gr.Textbox(label="Video Prompt", placeholder="e.g., The plague doctor's mask snaps open and closed, chanting")
+            
             with gr.Accordion("Settings", open=False):
                 sc = gr.Slider(0, 2, 0.6, label="Sketch Strictness")
                 cs = gr.Slider(0.5, 1, 0.85, label="Color Consistency")
+                
             btn = gr.Button("Generate Animation", variant="primary")
             
+        # --- OUTPUT COLUMN ---
         with gr.Column():
-            vid = gr.Video(label="Final Video")
+            # Frames moved to the top
             with gr.Row():
                 o1 = gr.Image(label="Start Frame")
                 o2 = gr.Image(label="End Frame")
+                
+            # Video moved below the frames
+            vid = gr.Video(label="Final Video")
 
-    btn.click(master_pipeline, [s1, s2, p, sc, cs], [o1, o2, vid])
+    # Updated click event to include both prompt inputs
+    btn.click(
+        fn=master_pipeline, 
+        inputs=[s1, s2, img_prompt, vid_prompt, sc, cs], 
+        outputs=[o1, o2, vid]
+    )
 
 demo.launch()