Update app.py

app.py

@@ -2,12 +2,17 @@ import os
 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
 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
+sys.path.insert(0, str(Path(__file__).parent))
+
 from demo_generate.halftone_method import apply_halftone
 
 # Environment setup
@@ -64,26 +69,6 @@ def remove_background(image_data):
         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():
@@ -118,10 +103,13 @@ def bg_remove():
             return jsonify({'error': 'Resim yüklenmedi'}), 400
             
         file = request.files['image']
-        if len(file.read()) > 10 * 1024 * 1024:
-            return jsonify({'error': 'Dosya boyutu 10MB üzerinde'}), 400
+        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())
         img_io = io.BytesIO()
         output.save(img_io, 'PNG')