Update app.py

app.py

@@ -26,17 +26,17 @@ def generate_jwt_token():
 
 def generate_image(image, prompt=""):
     """
-    Call Kling AI API to generate an image from an input image.
+    Call Kling AI API for single-reference face generation.
     
     Args:
-        image: Uploaded image file (from Gradio)
-        prompt (str): Optional text prompt to guide image transformation
+        image: Uploaded image file (from Gradio, face image)
+        prompt (str): Optional text prompt to guide transformation
     
     Returns:
         str: Path to the generated image or error message
     """
     if not image:
-        return "Error: Please upload a valid image (PNG/JPEG, <10 MB, ≥512x512 pixels)."
+        return "Error: Please upload a valid face image (PNG/JPEG, <10 MB, ≥512x512 pixels)."
     
     # Convert image to base64
     try:
@@ -52,8 +52,8 @@ def generate_image(image, prompt=""):
     
     payload = {
         "image": image_base64,
-        "prompt": prompt or "Transform the image into a cartoon style",
-        "strength": 0.7,  # Controls transformation intensity (0.0 to 1.0)
+        "prompt": prompt or "Transform the face into a cartoon style while preserving identity",
+        "strength": 0.97,  # High reference strength for face preservation
         "output_format": "png"
     }
     
@@ -85,7 +85,7 @@ def generate_image(image, prompt=""):
                     f.write(image_response.content)
                 return str(output_path)
             elif status == "failed":
-                return "Error: Image generation failed. Avoid NSFW, celebrities, or sensitive content (e.g., politics, violence)."
+                return "Error: Image generation failed. Ensure the image contains a clear face and avoid NSFW or sensitive content."
             elif status == "processing" and status_data.get("progress", 0) >= 0.99:
                 return "Error: Generation stuck at 99%. Check account credits or upgrade to a paid plan at https://klingai.com."
             time.sleep(5)
@@ -125,11 +125,11 @@ def generate_image(image, prompt=""):
             if service_code == 1200:
                 return "Error: Invalid request parameters (1200). Check payload format."
             elif service_code == 1201:
-                return "Error: Invalid parameters (1201). Use PNG/JPEG (<10 MB, ≥512x512) and a simple prompt."
+                return "Error: Invalid parameters (1201). Use PNG/JPEG (<10 MB, ≥512x512) with a clear face."
             elif service_code == 1300:
                 return "Error: Platform policy triggered (1300). Check input content."
             elif service_code == 1301:
-                return "Error: Content security policy triggered (1301). Avoid NSFW or sensitive content."
+                return "Error: Content security policy triggered (1301). Ensure the image contains a clear face and avoid NSFW content."
         elif status_code == 404:
             if service_code == 1202:
                 return "Error: Invalid request method (1202). Check https://klingai.com/global/dev for correct endpoint."
@@ -148,10 +148,10 @@ def generate_image(image, prompt=""):
 
 def chatbot_interface(image, prompt):
     """
-    Gradio interface for image-to-image generation.
+    Gradio interface for single-reference face generation.
     
     Args:
-        image: Uploaded image file
+        image: Uploaded image file (containing a face)
         prompt (str): Optional text prompt
     
     Returns:
@@ -163,12 +163,12 @@ def chatbot_interface(image, prompt):
 iface = gr.Interface(
     fn=chatbot_interface,
     inputs=[
-        gr.Image(type="filepath", label="Upload Image (PNG/JPEG, <10 MB, ≥512x512)"),
-        gr.Textbox(lines=2, placeholder="Enter an optional prompt (e.g., 'Turn this into a cartoon')", label="Prompt")
+        gr.Image(type="filepath", label="Upload Face Image (PNG/JPEG, <10 MB, ≥512x512)"),
+        gr.Textbox(lines=2, placeholder="Enter an optional prompt (e.g., 'Turn this face into a cartoon')", label="Prompt")
     ],
-    outputs=gr.Image(label="Generated Image"),
-    title="Kling AI Image-to-Image Generator",
-    description="Upload a PNG/JPEG image (<10 MB, ≥512x512 pixels) to generate a transformed image using Kling AI API. Avoid NSFW, celebrities, or sensitive content."
+    outputs=gr.Image(label="Generated Face Image"),
+    title="Kling AI Single-Reference Face Generator",
+    description="Upload a PNG/JPEG image (<10 MB, ≥512x512 pixels) with a clear face to generate a transformed image using Kling AI API (reference strength 0.97). Avoid NSFW or sensitive content."
 )
 
 # Launch the interface