Update app.py

app.py

@@ -39,23 +39,23 @@ def wait_for_image_with_prefix(folder, prefix):
         time.sleep(1)
         return initial_size == os.path.getsize(file_path)
 
-    while True:
-        files = os.listdir(folder)
-        image_files = [f for f in files if f.lower().startswith(prefix.lower()) and
-                       f.lower().endswith(('.png', '.jpg', '.jpeg'))]
-
-        if image_files:
-            # Sort by modification time to get the latest file
-            image_files.sort(key=lambda x: os.path.getmtime(os.path.join(folder, x)), reverse=True)
-            latest_image = os.path.join(folder, image_files[0])
-
-            if is_file_ready(latest_image):
-                # Wait a bit more to ensure the file is completely written
-                time.sleep(3)
-                return latest_image
-
-        # If no matching file found, wait before checking again
-        time.sleep(1)
+    
+    files = os.listdir(folder)
+    image_files = [f for f in files if f.lower().startswith(prefix.lower()) and
+                   f.lower().endswith(('.png', '.jpg', '.jpeg'))]
+
+    if image_files:
+        # Sort by modification time to get the latest file
+        image_files.sort(key=lambda x: os.path.getmtime(os.path.join(folder, x)), reverse=True)
+        latest_image = os.path.join(folder, image_files[0])
+
+        if is_file_ready(latest_image):
+            # Wait a bit more to ensure the file is completely written
+            time.sleep(3)
+            return latest_image
+
+    
+    return None
 
 
 def delete_image_file(file_path):