Update app.py

app.py

@@ -3,104 +3,46 @@ import sys
 import io
 import logging
 from pathlib import Path
-from datetime import datetime
 from flask import Flask, request, send_file, jsonify
-from PIL import Image, ImageDraw
+from PIL import Image
 import torch
-import torchvision.transforms as transforms
 from database import init_db, close_db, get_db_path
 
-# Proje kök dizinini Python path'ine ekle
+# Proje yolu ayarı
 sys.path.insert(0, str(Path(__file__).parent))
 
-from demo_generate.halftone_method import apply_halftone
-from demo_generate.animegan_method import apply_animegan  # Yeni eklenen
-from demo_generate.rembg_method import remove_background  # Yeni eklenen
+# Modül importları
+try:
+    from demo_generate.halftone_method import apply_halftone
+    from demo_generate.animegan_method import apply_animegan
+    from demo_generate.rembg_method import remove_background
+except ImportError as e:
+    logging.error(f"Modül import hatası: {e}")
+    raise
 
-# Environment setup
+# Ortam ayarları
 os.environ['NUMBA_DISABLE_JIT'] = '1'
 os.environ['TORCH_HOME'] = '/tmp/torch_cache'
 os.environ['U2NET_HOME'] = '/tmp/.u2net'
 
-# Create cache directories
+# Önbellek klasörleri
 os.makedirs('/tmp/torch_cache', exist_ok=True)
 os.makedirs('/tmp/.u2net', exist_ok=True)
 
-# Configure logging
+# Log ayarı
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Initialize Flask
+# Flask uygulaması
 app = Flask(__name__)
 init_db(app)
 
-# --- Model Initialization ---
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-
-# AnimeGAN model yükleme (orijinal kodunuz aynen korundu)
-try:
-    logger.info("Loading AnimeGAN model...")
-    model = torch.hub.load(
-        'bryandlee/animegan2-pytorch:main',
-        'generator',
-        pretrained='face_paint_512_v2',
-        trust_repo=True
-    ).to(device)
-    model.eval()
-    logger.info("Model loaded successfully")
-except Exception as e:
-    logger.error(f"Model loading failed: {str(e)}")
-    raise
-
-# --- Image Processing Functions ---
-def process_image(image):
-    """Convert image to anime style"""
-    transform = transforms.Compose([
-        transforms.Resize((512, 512)),
-        transforms.ToTensor(),
-        transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])
-    ])
-    with torch.no_grad():
-        output = model(transform(image).unsqueeze(0).to(device))
-    return transforms.ToPILImage()((output * 0.5 + 0.5).squeeze().cpu())
-
-def remove_background_original(image_data):  # İsim değiştirildi
-    """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)
-
-# --- API Routes ---
+# Ana sayfa
 @app.route('/')
 def home():
-    return "Anime Processing API - Türkçe"
-
-# Yeni eklenen endpoint: /generate-v2
-@app.route('/generate-v2', methods=['POST'])
-def generate_v2():
-    try:
-        if 'image' not in request.files:
-            return jsonify({'error': 'Resim yüklenmedi'}), 400
-            
-        file = request.files['image']
-        if not file.filename:
-            return jsonify({'error': 'Boş dosya'}), 400
+    return "Anime İşleme API - Çalışıyor"
 
-        image = Image.open(io.BytesIO(file.read())).convert("RGB")
-        processed_img = apply_animegan(image)  # Yeni modül kullanılıyor
-        
-        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
-
-# Orijinal endpoint'ler aynen korundu
+# Anime dönüştürme
 @app.route('/generate', methods=['POST'])
 def generate():
     try:
@@ -109,10 +51,10 @@ def generate():
             
         file = request.files['image']
         if not file.filename:
-            return jsonify({'error': 'Boş dosya'}), 400
+            return jsonify({'error': 'Geçersiz dosya'}), 400
 
         image = Image.open(io.BytesIO(file.read())).convert("RGB")
-        processed_img = process_image(image)
+        processed_img = apply_animegan(image)
         
         img_io = io.BytesIO()
         processed_img.save(img_io, 'PNG')
@@ -121,34 +63,9 @@ def generate():
         
     except Exception as e:
         logger.error(f"Hata: {str(e)}")
-        return jsonify({'error': str(e)}), 500
+        return jsonify({'error': 'İşlem sırasında hata oluştu'}), 500
 
-# Yeni eklenen endpoint: /remove-bg-v2
-@app.route('/remove-bg-v2', methods=['POST'])
-def bg_remove_v2():
-    try:
-        if 'image' not in request.files:
-            return jsonify({'error': 'Resim yüklenmedi'}), 400
-            
-        file = request.files['image']
-        file.seek(0, os.SEEK_END)
-        file_size = file.tell()
-        file.seek(0)
-        
-        if file_size > 10 * 1024 * 1024:
-            return jsonify({'error': 'Dosya boyutu 10MB üzerinde'}), 400
-        
-        output = remove_background(file.read())  # Yeni modül kullanılıyor
-        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
-
-# Orijinal endpoint aynen korundu
+# Arkaplan kaldırma
 @app.route('/remove-bg', methods=['POST'])
 def bg_remove():
     try:
@@ -156,14 +73,10 @@ def bg_remove():
             return jsonify({'error': 'Resim yüklenmedi'}), 400
             
         file = request.files['image']
-        file.seek(0, os.SEEK_END)
-        file_size = file.tell()
-        file.seek(0)
-        
-        if file_size > 10 * 1024 * 1024:
-            return jsonify({'error': 'Dosya boyutu 10MB üzerinde'}), 400
-        
-        output = remove_background_original(file.read())
+        if file.filename == '':
+            return jsonify({'error': 'Boş dosya'}), 400
+
+        output = remove_background(file.read())
         img_io = io.BytesIO()
         output.save(img_io, 'PNG')
         img_io.seek(0)
@@ -173,7 +86,7 @@ def bg_remove():
         logger.error(f"Arkaplan kaldırma hatası: {str(e)}")
         return jsonify({'error': str(e)}), 500
 
-# Orijinal halftone endpoint'i aynen korundu
+# Halftone efekti
 @app.route('/halftone', methods=['POST'])
 def halftone_route():
     try:
@@ -196,7 +109,7 @@ def halftone_route():
         logger.error(f"Halftone hatası: {str(e)}")
         return jsonify({'error': str(e)}), 500
 
-# --- Database Routes --- (Aynen korundu)
+# Veritabanı işlemleri
 @app.route('/users', methods=['GET'])
 def get_users_route():
     from get_methods import get_users
@@ -217,7 +130,7 @@ def consume_user_route(udid):
     from consume_method import consume_user
     return consume_user(udid)
 
-# --- Teardown --- (Aynen korundu)
+# Kapatma
 @app.teardown_appcontext
 def shutdown_session(exception=None):
     close_db()