Update app.py

app.py

@@ -10,6 +10,10 @@ 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.
@@ -23,28 +27,26 @@ 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:
-        # 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
+        # Run the image_to_animation.py script
         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")
         
@@ -58,14 +60,16 @@ def animate_image(file_path):
     except Exception as e:
         raise RuntimeError(f"Unexpected error: {e}")
 
-# Define the Gradio interface using the updated API
+# Create the Gradio interface
 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."
+    description="Upload your drawing or take a photo, and get an animated GIF.",
+    flagging_dir=FLAGGED_DIR
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    # Listen on 0.0.0.0:7860 (default for Hugging Face Spaces)
+    iface.launch(server_name="0.0.0.0", server_port=7860)