basics for demo done

FaceEnhancementProd.py

@@ -188,7 +188,7 @@ def main(
     output_image: str,
     dist_image: str = None,
     positive_prompt: str = "",
-    # models: dict = None
+    id_weight: float = 0.75,
 ):
     global models
     if models is None:
@@ -241,7 +241,7 @@ def main(
         saveimage = SaveImage()
 
         applypulidflux_133 = applypulidflux.apply_pulid_flux(
-            weight=0.7500000000000001,
+            weight=id_weight,
             start_at=0.10000000000000002,
             end_at=1,
             fusion="mean",
@@ -317,9 +317,9 @@ def save_comfy_images(images, output_dirs):
         pil_image.save(output_dirs[idx])
 
 
-def enhance_face(face_image: str, input_image: str, output_image: str, dist_image: str = None, positive_prompt: str = ""):
+def enhance_face(face_image: str, input_image: str, output_image: str, dist_image: str = None, positive_prompt: str = "", id_weight: float = 0.75):
     initialize_models()  # Ensure models are loaded
-    main(face_image, input_image, output_image, dist_image, positive_prompt)
+    main(face_image, input_image, output_image, dist_image, positive_prompt, id_weight)
 
 if __name__ == "__main__":
     pass

demo.py

@@ -4,6 +4,8 @@ import tempfile
 from main import process_face
 from PIL import Image
 
+PORT = 7860
+
 def enhance_face_gradio(input_image, ref_image):
     """
     Wrapper function for process_face that works with Gradio.
@@ -87,18 +89,25 @@ def create_gradio_interface():
         # Add examples using gr.Examples
         gr.Markdown("## Examples")
         example_inps = [
-            ["examples/chatgpt_dany_1.png", "examples/dany_face.jpg"],
-            ["examples/chatgpt_dany_2.png", "examples/dany_face.jpg"]
+            ["examples/dany_gpt_1.png", "examples/dany_face.jpg"],
+            ["examples/dany_gpt_2.png", "examples/dany_face.jpg"],
         ]
         gr.Examples(examples=example_inps, inputs=[input_image, ref_image], outputs=output_image)
 
     # Launch the Gradio app with queue
-    demo.queue(max_size=20)
-    demo.launch(
-        share=True,  # Set to True if you want a public link
-        server_name="0.0.0.0",  # Make available on all network interfaces
-        server_port=7860,
-    )
+    demo.queue(max_size=99)
+    
+    try:
+        demo.launch(
+            share=True, 
+            server_name="0.0.0.0",
+            server_port=PORT,
+            quiet=True,
+            show_error=True,
+        )
+    except OSError as e:
+        print(f"Error starting server: {e}")
+        sys.exit(1)
 
 
 if __name__ == "__main__":

main.py

@@ -83,7 +83,7 @@ def process_face(input_path, ref_path, crop=False, upscale=False, output_path=No
     comfy_ref_path = os.path.relpath(scratch_ref, "./ComfyUI/input")
     comfy_input_path = os.path.relpath(scratch_input, "./ComfyUI/input")
     
-    enhance_face(comfy_ref_path, comfy_input_path, output_path, dist_image=f"{output_path}_dist.png")
+    enhance_face(comfy_ref_path, comfy_input_path, output_path, dist_image=f"{output_path}_dist.png", id_weight=0.75)
     
     print(f"Enhanced image saved to: {output_path}")
     print(f"Working files are in: {scratch_dir}")