Update app.py

app.py

@@ -1,17 +1,36 @@
-from flask import Flask
+from flask import Flask, request, send_file, jsonify
 import os
 import sys
+import io
+import logging
 from database import init_db, close_db, get_db_path
 
 # Import yollarını kontrol et
 sys.path.append(os.path.dirname(os.path.abspath(__file__)))
 
+# Environment setup
+os.environ['NUMBA_DISABLE_JIT'] = '1'
+os.environ['TORCH_HOME'] = '/tmp/torch_cache'
+os.environ['U2NET_HOME'] = '/tmp/.u2net'
+
+# Create cache directories if they don't exist
+os.makedirs('/tmp/torch_cache', exist_ok=True)
+os.makedirs('/tmp/.u2net', exist_ok=True)
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# Import methods
 try:
     from get_methods import get_users, get_user
     from post_methods import add_user
     from consume_method import consume_user
+    from animegan_method import generate_anime
+    from rembg_method import remove_background
+    from halftone_method import apply_halftone
 except ImportError as e:
-    print(f"Import hatası: {e}")
+    logger.error(f"Import hatası: {e}")
     raise
 
 app = Flask(__name__)
@@ -21,11 +40,80 @@ init_db(app)
 def home():
     return "StableDiffusionAPI Türkçe"
 
+# Existing routes
 app.route('/users', methods=['GET'])(get_users)
 app.route('/user/<string:udid>', methods=['GET'])(get_user)
 app.route('/add-user', methods=['POST'])(add_user)
 app.route('/consume/<string:udid>', methods=['POST'])(consume_user)
 
+# New image processing routes
+@app.route('/generate', methods=['POST'])
+def generate():
+    try:
+        if 'image' not in request.files:
+            logger.error("No file provided")
+            return jsonify({'error': 'No image provided'}), 400
+            
+        file = request.files['image']
+        if file.filename == '':
+            logger.error("Empty file provided")
+            return jsonify({'error': 'No selected file'}), 400
+            
+        img_io = generate_anime(file.read())
+        return send_file(img_io, mimetype='image/png')
+    except Exception as e:
+        logger.error(f"Unexpected error: {str(e)}", exc_info=True)
+        return jsonify({'error': str(e)}), 500
+
+@app.route('/remove-bg', methods=['POST'])
+def remove_bg():
+    try:
+        if 'image' not in request.files:
+            logger.error("No file provided")
+            return jsonify({'error': 'No image provided'}), 400
+            
+        file = request.files['image']
+        if file.filename == '':
+            logger.error("Empty file provided")
+            return jsonify({'error': 'No selected file'}), 400
+            
+        max_size = 10 * 1024 * 1024  # 10MB
+        file_data = file.read()
+        if len(file_data) > max_size:
+            logger.error(f"File too large: {len(file_data)} bytes")
+            return jsonify({'error': 'File too large (max 10MB)'}), 400
+            
+        img_io = remove_background(file_data)
+        return send_file(img_io, mimetype='image/png')
+    except Exception as e:
+        logger.error(f"Unexpected error: {str(e)}", exc_info=True)
+        return jsonify({'error': str(e)}), 500
+
+@app.route('/halftone', methods=['POST'])
+def halftone():
+    try:
+        if 'image' not in request.files:
+            logger.error("No file provided")
+            return jsonify({'error': 'No image provided'}), 400
+            
+        file = request.files['image']
+        if file.filename == '':
+            logger.error("Empty file provided")
+            return jsonify({'error': 'No selected file'}), 400
+            
+        max_size = 10 * 1024 * 1024  # 10MB
+        file_data = file.read()
+        if len(file_data) > max_size:
+            logger.error(f"File too large: {len(file_data)} bytes")
+            return jsonify({'error': 'File too large (max 10MB)'}), 400
+            
+        dot_size = int(request.form.get('dot_size', 10))
+        img_io = apply_halftone(file_data, dot_size)
+        return send_file(img_io, mimetype='image/png')
+    except Exception as e:
+        logger.error(f"Unexpected error: {str(e)}", exc_info=True)
+        return jsonify({'error': str(e)}), 500
+
 @app.teardown_appcontext
 def shutdown_session(exception=None):
     close_db()