Update app.py

app.py

@@ -147,10 +147,19 @@ def process_image(image_input):
     # Initialize the Gradio client with the URL of the Gradio server
     client = Client("https://adept-fuyu-8b-demo.hf.space/--replicas/pqjvl/")
 
-    # Check if the image input is a file path (str) or a PIL Image
+    # Check if the image input is a base64 string, file path (str), or a PIL Image
     if isinstance(image_input, str):
-        # Assuming it's a file path or a URL
-        image_path = image_input
+        try:
+            # Try to decode if it's a base64 string
+            image = decode_image(image_input)
+        except Exception:
+            # If decoding fails, assume it's a file path or a URL
+            image_path = image_input
+        else:
+            # If decoding succeeds, save the decoded image to a temporary file
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp_file:
+                image.save(tmp_file.name)
+                image_path = tmp_file.name
     else:
         # Assuming it's a PIL Image, save it to a temporary file
         with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp_file:
@@ -165,7 +174,7 @@ def process_image(image_input):
     )
 
     # Clean up the temporary file if created
-    if not isinstance(image_input, str):
+    if not isinstance(image_input, str) or isinstance(image_input, str) and 'tmp' in image_path:
         os.remove(image_path)
 
     return result