Update app.py

app.py

@@ -14,6 +14,8 @@ from database import init_db, close_db, get_db_path
 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
 
 # Environment setup
 os.environ['NUMBA_DISABLE_JIT'] = '1'
@@ -35,6 +37,7 @@ 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(
@@ -61,7 +64,7 @@ def process_image(image):
         output = model(transform(image).unsqueeze(0).to(device))
     return transforms.ToPILImage()((output * 0.5 + 0.5).squeeze().cpu())
 
-def remove_background(image_data):
+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))
@@ -74,6 +77,30 @@ def remove_background(image_data):
 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
+
+        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
 @app.route('/generate', methods=['POST'])
 def generate():
     try:
@@ -96,6 +123,32 @@ def generate():
         logger.error(f"Hata: {str(e)}")
         return jsonify({'error': str(e)}), 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
 @app.route('/remove-bg', methods=['POST'])
 def bg_remove():
     try:
@@ -110,7 +163,7 @@ def bg_remove():
         if file_size > 10 * 1024 * 1024:
             return jsonify({'error': 'Dosya boyutu 10MB üzerinde'}), 400
         
-        output = remove_background(file.read())
+        output = remove_background_original(file.read())
         img_io = io.BytesIO()
         output.save(img_io, 'PNG')
         img_io.seek(0)
@@ -120,6 +173,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
 @app.route('/halftone', methods=['POST'])
 def halftone_route():
     try:
@@ -142,7 +196,7 @@ def halftone_route():
         logger.error(f"Halftone hatası: {str(e)}")
         return jsonify({'error': str(e)}), 500
 
-# --- Database Routes ---
+# --- Database Routes --- (Aynen korundu)
 @app.route('/users', methods=['GET'])
 def get_users_route():
     from get_methods import get_users
@@ -163,7 +217,7 @@ def consume_user_route(udid):
     from consume_method import consume_user
     return consume_user(udid)
 
-# --- Teardown ---
+# --- Teardown --- (Aynen korundu)
 @app.teardown_appcontext
 def shutdown_session(exception=None):
     close_db()