Update app.py

app.py

@@ -36,7 +36,7 @@ print(device)
 def wait_for_image_with_prefix(folder, prefix):
     def is_file_ready(file_path):
         initial_size = os.path.getsize(file_path)
-        time.sleep(0.5)
+        time.sleep(1)
         return initial_size == os.path.getsize(file_path)
 
     while True:
@@ -51,7 +51,7 @@ def wait_for_image_with_prefix(folder, prefix):
 
             if is_file_ready(latest_image):
                 # Wait a bit more to ensure the file is completely written
-                time.sleep(1)
+                time.sleep(3)
                 return latest_image
 
         # If no matching file found, wait before checking again
@@ -133,9 +133,12 @@ def generate_image(prompt, image):
     finally:        
         if process and process.poll() is None:
             process.terminate()
+            logger.debug("process.terminate()")
             try:
+                logger.debug("process.wait(timeout=5)")
                 process.wait(timeout=5)
             except subprocess.TimeoutExpired:
+                logger.debug("process.kill()")
                 process.kill()