caching images with hash

.gitignore

@@ -3,4 +3,5 @@ venv
 .idea/
 .gradio/
 __pycache__/
-ComfyUI/
+ComfyUI/
+cache/

README.md

@@ -42,9 +42,9 @@ A tool for improving facial consistency and quality in AI-generated images. Dram
 
 4. Run inference on one example:
 
-```
-python main.py --input examples/dany_gpt_1.png --ref examples/dany_face.jpg --out examples/dany_enhanced.png
-```
+   ```
+   python main.py --input examples/dany_gpt_1.png --ref examples/dany_face.jpg --out examples/dany_enhanced.png
+   ```
 
 ## Running on ComfyUI
 
@@ -74,7 +74,7 @@ A simple web interface for the face enhancement workflow.
 ### Notes
 - The script and demo run a ComfyUI server ephemerally
 - Gradio demo is faster than the script because models remain loaded in memory
-- All images are saved in ./ComfyUI/input/scratch/
+- All images are saved in `./ComfyUI/input/scratch/`
 - Temporary files are created during processing and cleaned up afterward
 
 ### Troubleshooting

main.py

@@ -50,13 +50,6 @@ def create_scratch_dir():
 def process_face(input_path, ref_path, crop=False, upscale=False, output_path=None, id_weight=0.75):
     """
     Process a face image using the given parameters.
-    
-    Args:
-        input_path (str): Path to the input image
-        ref_path (str): Path to the reference image
-        crop (bool): Whether to crop the image
-        upscale (bool): Whether to upscale the image
-        output_path (str): Path to save the output image
         
     Returns:
         str: Path to the scratch directory used for processing