Update app.py

app.py

@@ -10,10 +10,6 @@ OUTPUT_FOLDER = 'outputs_gradio'
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 os.makedirs(OUTPUT_FOLDER, exist_ok=True)
 
-# Create flagged directory inside outputs_gradio to avoid permission issues
-FLAGGED_DIR = os.path.join(OUTPUT_FOLDER, 'flagged')
-os.makedirs(FLAGGED_DIR, exist_ok=True)
-
 def animate_image(file_path):
     """
     Function to process the uploaded image and generate an animated GIF.
@@ -27,26 +23,28 @@ def animate_image(file_path):
     if not file_path:
         raise ValueError("No file uploaded.")
 
+    # 'file_path' is a string path to the uploaded file
     input_path = file_path
     filename = os.path.basename(input_path)
     base, ext = os.path.splitext(filename)
     
-    # Validate file extension
-    allowed_extensions = ['.png', '.jpg', '.jpeg', '.bmp']
-    if ext.lower() not in allowed_extensions:
-        raise ValueError("Unsupported file type. Please upload an image file (png, jpg, jpeg, bmp).")
-
     # Define the annotation directory for this specific image
     char_anno_dir = os.path.join(OUTPUT_FOLDER, f"{base}_out")
     os.makedirs(char_anno_dir, exist_ok=True)
     
     try:
-        # Run the image_to_animation.py script
+        # Validate file extension
+        allowed_extensions = ['.png', '.jpg', '.jpeg', '.bmp']
+        if ext.lower() not in allowed_extensions:
+            raise ValueError("Unsupported file type. Please upload an image file (png, jpg, jpeg, bmp).")
+        
+        # Run the image_to_animation.py script with required arguments
         subprocess.run([
             'python', 'examples/image_to_animation.py',
             input_path, char_anno_dir
+            # Optionally, add motion_cfg_fn and retarget_cfg_fn here if needed
         ], check=True)
-
+        
         # Path to the generated GIF
         gif_path = os.path.join(char_anno_dir, "video.gif")
         
@@ -60,16 +58,14 @@ def animate_image(file_path):
     except Exception as e:
         raise RuntimeError(f"Unexpected error: {e}")
 
-# Create the Gradio interface
+# Define the Gradio interface using the updated API
 iface = gr.Interface(
     fn=animate_image,
     inputs=gr.Image(label="Upload Drawing", type="filepath", sources=["upload", "webcam"]),
     outputs=gr.Image(label="Animated GIF"),
     title="Animated Drawings",
-    description="Upload your drawing or take a photo, and get an animated GIF.",
-    flagging_dir=FLAGGED_DIR
+    description="Upload your drawing or take a photo, and get an animated GIF."
 )
 
 if __name__ == "__main__":
-    # Listen on 0.0.0.0:7860 (default for Hugging Face Spaces)
-    iface.launch(server_name="0.0.0.0", server_port=7860)
+    iface.launch()