Update app.py

app.py

@@ -149,14 +149,14 @@ def generate_image(subject_images, prompt):
     return None, "Task timed out after 10 minutes"
 
 # ===== GRADIO INTERFACE =====
-def process_interface(subject_images, prompt):
-    # Filter out None values from subject_images
-    valid_images = [img for img in subject_images if img is not None]
+def process_interface(subject_image1, subject_image2, subject_image3, subject_image4, prompt):
+    # Filter out None/empty images
+    subject_images = [img for img in [subject_image1, subject_image2, subject_image3, subject_image4] if img]
     
-    if len(valid_images) < 2:
+    if len(subject_images) < 2:
         return None, None, "Please upload at least 2 subject images"
     
-    output_path, error = generate_image(valid_images, prompt)
+    output_path, error = generate_image(subject_images, prompt)
     if error:
         return None, None, error
     
@@ -170,24 +170,24 @@ with gr.Blocks(title="Kling AI Multi-Image Generator") as app:
         with gr.Column():
             gr.Markdown("### Input Settings")
             with gr.Row():
-                subject_image1 = gr.Image(type="filepath", label="Subject Image 1")
-                subject_image2 = gr.Image(type="filepath", label="Subject Image 2")
+                subject_image1 = gr.Image(type="filepath", label="Subject Image 1 *")
+                subject_image2 = gr.Image(type="filepath", label="Subject Image 2 *")
             with gr.Row():
                 subject_image3 = gr.Image(type="filepath", label="Subject Image 3 (Optional)")
                 subject_image4 = gr.Image(type="filepath", label="Subject Image 4 (Optional)")
             
             prompt_input = gr.Textbox(
                 label="Transformation Prompt",
-                placeholder="Describe how to combine these images (e.g. 'merge facial features from all images')"
+                placeholder="Describe how to combine these images"
             )
             
             generate_btn = gr.Button("Generate", variant="primary")
             
-            gr.Markdown("### Requirements")
+            gr.Markdown("### Requirements (* = required)")
             gr.Markdown("""
-            - **Minimum 2 subject images required**
+            - **At least 2 subject images** (marked with *)
             - Max 4 images total
-            - Max size per image: 10MB
+            - Max size: 10MB per image
             - Formats: JPG, PNG
             - Min dimensions: 300x300px
             """)
@@ -195,17 +195,13 @@ with gr.Blocks(title="Kling AI Multi-Image Generator") as app:
         with gr.Column():
             gr.Markdown("### Output")
             output_image = gr.Image(label="Generated Image", interactive=False, height=400)
-            output_file = gr.File(label="Download Result", file_types=["image/png"])
+            output_file = gr.File(label="Download Result")
             status_output = gr.Textbox(label="Status", interactive=False)
     
+    # Modified inputs to accept individual components
     generate_btn.click(
         fn=process_interface,
-        inputs=[gr.components.List([
-            subject_image1, 
-            subject_image2, 
-            subject_image3, 
-            subject_image4
-        ]), prompt_input],
+        inputs=[subject_image1, subject_image2, subject_image3, subject_image4, prompt_input],
         outputs=[output_image, output_file, status_output]
     )