Upload app.py with huggingface_hub

app.py

@@ -0,0 +1,67 @@
+from flask import Flask, request, jsonify, send_from_directory
+from flask_cors import CORS
+import os
+import base64
+from colorizer import ImageColorizer
+
+# Configure Flask to serve the React build
+app = Flask(__name__, static_folder='frontend/dist', static_url_path='')
+CORS(app) 
+
+# Initialize colorizer
+colorizer = ImageColorizer()
+
+@app.route('/')
+def serve():
+    """Serve the React frontend."""
+    return send_from_directory(app.static_folder, 'index.html')
+
+@app.route('/api/status', methods=['GET'])
+def check_status():
+    """Check if model files are ready."""
+    try:
+        ready = colorizer.ensure_models()
+        return jsonify({"ready": ready})
+    except Exception as e:
+        return jsonify({"ready": False, "error": str(e)}), 500
+
+@app.route('/api/colorize', methods=['POST'])
+def colorize_image():
+    """Upload and colorize an image."""
+    if 'image' not in request.files:
+        return jsonify({"error": "No image uploaded"}), 400
+    
+    file = request.files['image']
+    if file.filename == '':
+        return jsonify({"error": "No selected file"}), 400
+    
+    try:
+        print(f"Processing image: {file.filename}")
+        img_bytes = file.read()
+        colorized_bytes = colorizer.colorize(img_bytes)
+        
+        # Return as base64 for easy frontend consumption
+        encoded = base64.b64encode(colorized_bytes).decode('utf-8')
+        return jsonify({
+            "image": f"data:image/jpeg;base64,{encoded}",
+            "message": "Successfully colorized"
+        })
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+        return jsonify({"error": str(e)}), 500
+
+# Error handler to serve index.html for React routes
+@app.errorhandler(404)
+def not_found(e):
+    return send_from_directory(app.static_folder, 'index.html')
+
+if __name__ == '__main__':
+    # Initial status check
+    print("Checking model files...")
+    colorizer.ensure_models()
+    
+    # Use environment variable for port (required for cloud deployment)
+    port = int(os.environ.get("PORT", 7860))
+    print(f"Starting server on port {port}...")
+    app.run(host='0.0.0.0', port=port)