Update app.py

app.py

@@ -1,57 +1,70 @@
+import gradio as gr
 import os
 import subprocess
-from flask import Flask, request, send_file, jsonify
-import uuid
-
-app = Flask(__name__)
 
 # Define directories for uploads and outputs
 UPLOAD_FOLDER = 'uploads_gradio'
 OUTPUT_FOLDER = 'outputs_gradio'
 
+# Create directories if they don't exist
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 os.makedirs(OUTPUT_FOLDER, exist_ok=True)
 
-@app.route('/animate', methods=['POST'])
-def animate():
-    if 'image' not in request.files:
-        return jsonify({"error": "No image file provided."}), 400
-
-    image_file = request.files['image']
-    if image_file.filename == '':
-        return jsonify({"error": "Empty filename."}), 400
-
-    # Save uploaded image
-    unique_id = str(uuid.uuid4())
-    input_path = os.path.join(UPLOAD_FOLDER, unique_id + os.path.splitext(image_file.filename)[1])
-    image_file.save(input_path)
+def animate_image(file_path):
+    """
+    Function to process the uploaded image and generate an animated GIF.
+    
+    Args:
+        file_path (str): Path to the uploaded file.
+    
+    Returns:
+        str: Path to the generated GIF.
+    """
+    if not file_path:
+        raise ValueError("No file uploaded.")
 
-    # Prepare output directory
-    char_anno_dir = os.path.join(OUTPUT_FOLDER, f"{unique_id}_out")
+    input_path = file_path
+    filename = os.path.basename(input_path)
+    base, ext = os.path.splitext(filename)
+    
+    # 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
         ], check=True)
-
+        
         # Path to the generated GIF
         gif_path = os.path.join(char_anno_dir, "video.gif")
-
-        if not os.path.exists(gif_path):
-            return jsonify({"error": "Animation not generated. Ensure input image is a clear humanoid drawing."}), 500
-
-        # Return the GIF as a response
-        # After returning, we can clean up if desired
-        return send_file(gif_path, mimetype="image/gif")
-
+        
+        if os.path.exists(gif_path):
+            return gif_path
+        else:
+            raise FileNotFoundError("Animation failed to generate. Please ensure the input image contains clear humanoid drawings.")
+    
     except subprocess.CalledProcessError as e:
-        return jsonify({"error": "Error during processing", "details": str(e)}), 500
+        raise RuntimeError(f"Error during processing: {e}")
     except Exception as e:
-        return jsonify({"error": "Unexpected error", "details": str(e)}), 500
+        raise RuntimeError(f"Unexpected error: {e}")
+
+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."
+)
 
 if __name__ == "__main__":
-    # Run the Flask app
-    app.run(host="0.0.0.0", port=7860, debug=False)
+    # Use the PORT environment variable provided by Hugging Face Spaces
+    port = int(os.getenv("PORT", "7860"))
+    iface.launch(server_name="0.0.0.0", server_port=port)