yusufgundogdu commited on
Commit
c92c276
·
verified ·
1 Parent(s): b2c5af8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -17
app.py CHANGED
@@ -5,20 +5,15 @@ import logging
5
  from pathlib import Path
6
  from flask import Flask, request, send_file, jsonify
7
  from PIL import Image
8
- import torch
9
  from database import init_db, close_db, get_db_path
10
 
11
  # Proje yolu ayarı
12
  sys.path.insert(0, str(Path(__file__).parent))
13
 
14
  # Modül importları
15
- try:
16
- from demo_generate.halftone_method import apply_halftone
17
- from demo_generate.animegan_method import apply_animegan
18
- from demo_generate.rembg_method import remove_background
19
- except ImportError as e:
20
- logging.error(f"Modül import hatası: {e}")
21
- raise
22
 
23
  # Ortam ayarları
24
  os.environ['NUMBA_DISABLE_JIT'] = '1'
@@ -42,7 +37,7 @@ init_db(app)
42
  def home():
43
  return "Anime İşleme API - Çalışıyor"
44
 
45
- # Anime dönüştürme
46
  @app.route('/generate', methods=['POST'])
47
  def generate():
48
  try:
@@ -53,19 +48,14 @@ def generate():
53
  if not file.filename:
54
  return jsonify({'error': 'Geçersiz dosya'}), 400
55
 
56
- image = Image.open(io.BytesIO(file.read())).convert("RGB")
57
- processed_img = apply_animegan(image)
58
-
59
- img_io = io.BytesIO()
60
- processed_img.save(img_io, 'PNG')
61
- img_io.seek(0)
62
  return send_file(img_io, mimetype='image/png')
63
 
64
  except Exception as e:
65
  logger.error(f"Hata: {str(e)}")
66
  return jsonify({'error': 'İşlem sırasında hata oluştu'}), 500
67
 
68
- # Arkaplan kaldırma
69
  @app.route('/remove-bg', methods=['POST'])
70
  def bg_remove():
71
  try:
@@ -86,7 +76,7 @@ def bg_remove():
86
  logger.error(f"Arkaplan kaldırma hatası: {str(e)}")
87
  return jsonify({'error': str(e)}), 500
88
 
89
- # Halftone efekti
90
  @app.route('/halftone', methods=['POST'])
91
  def halftone_route():
92
  try:
 
5
  from pathlib import Path
6
  from flask import Flask, request, send_file, jsonify
7
  from PIL import Image
 
8
  from database import init_db, close_db, get_db_path
9
 
10
  # Proje yolu ayarı
11
  sys.path.insert(0, str(Path(__file__).parent))
12
 
13
  # Modül importları
14
+ from demo_generate.halftone_method import apply_halftone
15
+ from demo_generate.animegan_method import generate_anime # AnimeGAN fonksiyonu
16
+ from demo_generate.rembg_method import remove_background
 
 
 
 
17
 
18
  # Ortam ayarları
19
  os.environ['NUMBA_DISABLE_JIT'] = '1'
 
37
  def home():
38
  return "Anime İşleme API - Çalışıyor"
39
 
40
+ # Anime dönüştürme endpoint'i
41
  @app.route('/generate', methods=['POST'])
42
  def generate():
43
  try:
 
48
  if not file.filename:
49
  return jsonify({'error': 'Geçersiz dosya'}), 400
50
 
51
+ img_io = generate_anime(file.read()) # AnimeGAN işlemi
 
 
 
 
 
52
  return send_file(img_io, mimetype='image/png')
53
 
54
  except Exception as e:
55
  logger.error(f"Hata: {str(e)}")
56
  return jsonify({'error': 'İşlem sırasında hata oluştu'}), 500
57
 
58
+ # Arkaplan kaldırma endpoint'i
59
  @app.route('/remove-bg', methods=['POST'])
60
  def bg_remove():
61
  try:
 
76
  logger.error(f"Arkaplan kaldırma hatası: {str(e)}")
77
  return jsonify({'error': str(e)}), 500
78
 
79
+ # Halftone efekti endpoint'i
80
  @app.route('/halftone', methods=['POST'])
81
  def halftone_route():
82
  try: