Update app.py

app.py

@@ -1,8 +1,13 @@
 import os
 import subprocess
+import logging
 from flask import Flask, request, send_file, jsonify
 from werkzeug.utils import secure_filename
-from flask_cors import CORS  # Import CORS for handling cross-origin requests
+from flask_cors import CORS
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 # Define directories for uploads and outputs
 UPLOAD_FOLDER = 'uploads_gradio'
@@ -53,6 +58,8 @@ def animate_image(file_path):
         raise ValueError("Unsupported file type. Please upload an image file (png, jpg, jpeg, bmp).")
 
     try:
+        logger.info(f"Starting animation for {input_path}")
+
         # Run the image_to_animation.py script with required arguments
         subprocess.run([
             'python', 'examples/image_to_animation.py',
@@ -63,13 +70,16 @@ def animate_image(file_path):
         gif_path = os.path.join(char_anno_dir, "video.gif")
 
         if os.path.exists(gif_path):
+            logger.info(f"Animation successful: {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:
+        logger.error(f"Error during processing: {e}")
         raise RuntimeError(f"Error during processing: {e}")
     except Exception as e:
+        logger.error(f"Unexpected error: {e}")
         raise RuntimeError(f"Unexpected error: {e}")
 
 @app.route('/animate', methods=['POST'])
@@ -77,30 +87,37 @@ def animate():
     """
     Endpoint to receive an image and return the animated GIF.
     """
+    logger.info("Received request to /animate")
     if 'file' not in request.files:
+        logger.warning("No file part in the request.")
         return jsonify({"error": "No file part in the request."}), 400
 
     file = request.files['file']
 
     if file.filename == '':
+        logger.warning("No file selected for uploading.")
         return jsonify({"error": "No file selected for uploading."}), 400
 
     if file and allowed_file(file.filename):
         filename = secure_filename(file.filename)
         input_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
         file.save(input_path)
+        logger.info(f"File saved to {input_path}")
 
         try:
             gif_path = animate_image(input_path)
+            logger.info(f"Sending GIF: {gif_path}")
             return send_file(
                 gif_path,
                 mimetype='image/gif',
                 as_attachment=True,
-                attachment_filename=f"{os.path.splitext(filename)[0]}.gif"
+                download_name=f"{os.path.splitext(filename)[0]}.gif"
             )
         except Exception as e:
+            logger.error(f"Error in /animate: {e}")
             return jsonify({"error": str(e)}), 500
     else:
+        logger.warning("Allowed file types are png, jpg, jpeg, bmp.")
         return jsonify({"error": "Allowed file types are png, jpg, jpeg, bmp."}), 400
 
 @app.route('/', methods=['GET'])
@@ -108,6 +125,7 @@ def index():
     """
     Root endpoint to provide basic information.
     """
+    logger.info("Received request to /")
     return jsonify({
         "message": "Animated Drawings API",
         "endpoints": {
@@ -120,9 +138,11 @@ def health():
     """
     Health check endpoint.
     """
+    logger.info("Received request to /health")
     return jsonify({"status": "healthy"}), 200
 
 if __name__ == '__main__':
     # Use the PORT environment variable provided by Hugging Face Spaces or default to 7860
     port = int(os.getenv("PORT", "7860"))
+    logger.info(f"Starting Flask app on port {port}")
     app.run(host='0.0.0.0', port=port)