Update app.py

app.py

@@ -4,7 +4,7 @@ import io
 import logging
 from datetime import datetime
 from flask import Flask, request, send_file, jsonify
-from PIL import Image
+from PIL import Image, ImageDraw
 import torch
 import torchvision.transforms as transforms
 from database import init_db, close_db, get_db_path
@@ -26,7 +26,7 @@ logger = logging.getLogger(__name__)
 app = Flask(__name__)
 init_db(app)
 
-# --- AnimeGAN Model Initialization ---
+# --- Model Initialization ---
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 try:
@@ -34,7 +34,7 @@ try:
     model = torch.hub.load(
         'bryandlee/animegan2-pytorch:main',
         'generator',
-        pretrained='face_paint_512_v2',  # Critical parameter
+        pretrained='face_paint_512_v2',
         trust_repo=True
     ).to(device)
     model.eval()
@@ -45,6 +45,7 @@ except Exception as e:
 
 # --- Image Processing Functions ---
 def process_image(image):
+    """Convert image to anime style"""
     transform = transforms.Compose([
         transforms.Resize((512, 512)),
         transforms.ToTensor(),
@@ -54,62 +55,122 @@ def process_image(image):
         output = model(transform(image).unsqueeze(0).to(device))
     return transforms.ToPILImage()((output * 0.5 + 0.5).squeeze().cpu())
 
-# --- Flask Routes ---
+def remove_background(image_data):
+    """Remove image background"""
+    from rembg import remove
+    input_image = Image.open(io.BytesIO(image_data))
+    if input_image.mode != 'RGB':
+        input_image = input_image.convert('RGB')
+    return remove(input_image)
+
+def apply_halftone(image, dot_size=10):
+    """Apply halftone effect"""
+    grayscale = image.convert('L')
+    width, height = grayscale.size
+    halftone_img = Image.new('L', (width, height), 255)
+    draw = ImageDraw.Draw(halftone_img)
+    
+    for x in range(0, width, dot_size):
+        for y in range(0, height, dot_size):
+            block = grayscale.crop((x, y, x+dot_size, y+dot_size))
+            avg = sum(block.getdata()) / (dot_size**2)
+            radius = int((255 - avg) / 255 * (dot_size//2))
+            draw.ellipse([
+                x+dot_size//2-radius, 
+                y+dot_size//2-radius,
+                x+dot_size//2+radius,
+                y+dot_size//2+radius
+            ], fill=0)
+    return halftone_img
+
+# --- API Routes ---
 @app.route('/')
 def home():
-    return "AnimeGAN API - Türkçe"
+    return "Anime Processing API - Türkçe"
 
 @app.route('/generate', methods=['POST'])
 def generate():
-    start_time = datetime.now()
-    logger.info(f"Generate request started at {start_time}")
-    
     try:
         if 'image' not in request.files:
-            return jsonify({'error': 'No image provided'}), 400
+            return jsonify({'error': 'Resim yüklenmedi'}), 400
             
         file = request.files['image']
         if not file.filename:
-            return jsonify({'error': 'Empty filename'}), 400
+            return jsonify({'error': 'Boş dosya'}), 400
+
+        image = Image.open(io.BytesIO(file.read())).convert("RGB")
+        processed_img = process_image(image)
+        
+        img_io = io.BytesIO()
+        processed_img.save(img_io, 'PNG')
+        img_io.seek(0)
+        return send_file(img_io, mimetype='image/png')
+        
+    except Exception as e:
+        logger.error(f"Hata: {str(e)}")
+        return jsonify({'error': str(e)}), 500
 
-        try:
-            image = Image.open(io.BytesIO(file.read())).convert("RGB")
-            processed_img = process_image(image)
-            
-            img_io = io.BytesIO()
-            processed_img.save(img_io, 'PNG')
-            img_io.seek(0)
-            
-            duration = (datetime.now() - start_time).total_seconds()
-            logger.info(f"Processed in {duration:.2f} seconds")
-            return send_file(img_io, mimetype='image/png')
+@app.route('/remove-bg', methods=['POST'])
+def bg_remove():
+    try:
+        if 'image' not in request.files:
+            return jsonify({'error': 'Resim yüklenmedi'}), 400
             
-        except Exception as e:
-            logger.error(f"Processing error: {str(e)}")
-            return jsonify({'error': str(e)}), 500
+        file = request.files['image']
+        if len(file.read()) > 10 * 1024 * 1024:
+            return jsonify({'error': 'Dosya boyutu 10MB üzerinde'}), 400
+        file.seek(0)
+        
+        output = remove_background(file.read())
+        img_io = io.BytesIO()
+        output.save(img_io, 'PNG')
+        img_io.seek(0)
+        return send_file(img_io, mimetype='image/png')
+        
+    except Exception as e:
+        logger.error(f"Arkaplan kaldırma hatası: {str(e)}")
+        return jsonify({'error': str(e)}), 500
+
+@app.route('/halftone', methods=['POST'])
+def halftone_route():
+    try:
+        if 'image' not in request.files:
+            return jsonify({'error': 'Resim yüklenmedi'}), 400
             
+        file = request.files['image']
+        dot_size = int(request.form.get('dot_size', 10))
+        dot_size = max(5, min(20, dot_size))
+        
+        image = Image.open(io.BytesIO(file.read()))
+        output = apply_halftone(image, dot_size)
+        
+        img_io = io.BytesIO()
+        output.save(img_io, 'PNG')
+        img_io.seek(0)
+        return send_file(img_io, mimetype='image/png')
+        
     except Exception as e:
-        logger.error(f"Unexpected error: {str(e)}")
-        return jsonify({'error': 'Internal server error'}), 500
+        logger.error(f"Halftone hatası: {str(e)}")
+        return jsonify({'error': str(e)}), 500
 
-# --- Database Routes (Keep your existing routes) ---
+# --- Database Routes ---
 @app.route('/users', methods=['GET'])
-def get_users():
+def get_users_route():
     from get_methods import get_users
     return get_users()
 
 @app.route('/user/<string:udid>', methods=['GET'])
-def get_user(udid):
+def get_user_route(udid):
     from get_methods import get_user
     return get_user(udid)
 
 @app.route('/add-user', methods=['POST'])
-def add_user():
+def add_user_route():
     from post_methods import add_user
     return add_user()
 
 @app.route('/consume/<string:udid>', methods=['POST'])
-def consume_user(udid):
+def consume_user_route(udid):
     from consume_method import consume_user
     return consume_user(udid)