Update app.py

app.py

@@ -1,70 +1,57 @@
-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)
 
-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.")
+@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)
 
-    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")
+    # Prepare output directory
+    char_anno_dir = os.path.join(OUTPUT_FOLDER, f"{unique_id}_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
         ], check=True)
-        
+
         # Path to the generated GIF
         gif_path = os.path.join(char_anno_dir, "video.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.")
-    
+
+        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")
+
     except subprocess.CalledProcessError as e:
-        raise RuntimeError(f"Error during processing: {e}")
+        return jsonify({"error": "Error during processing", "details": str(e)}), 500
     except Exception as e:
-        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."
-)
+        return jsonify({"error": "Unexpected error", "details": str(e)}), 500
 
 if __name__ == "__main__":
-    # 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)
+    # Run the Flask app
+    app.run(host="0.0.0.0", port=7860, debug=False)