Update app.py

app.py

@@ -1,149 +1,63 @@
-import os
-import sys
-import io
-import logging
-from pathlib import Path
-from flask import Flask, request, send_file, jsonify, send_from_directory
-from PIL import Image
-from database import init_db, close_db, get_db, get_generate_list, add_to_generate_list, get_db_path
+app.py
+from flask import Flask, request, jsonify
+import numpy as np
+import random
 
-# Proje yolu ayarı
-sys.path.insert(0, str(Path(__file__).parent))
-
-# Modül importları
-from demo_generate.halftone_method import apply_halftone
-from demo_generate.animegan_method import generate_anime
-from demo_generate.rembg_method import remove_background
-
-# Ortam ayarları
-os.environ['NUMBA_DISABLE_JIT'] = '1'
-os.environ['TORCH_HOME'] = '/tmp/torch_cache'
-os.environ['U2NET_HOME'] = '/tmp/.u2net'
-
-# Önbellek klasörleri
-os.makedirs('/tmp/torch_cache', exist_ok=True)
-os.makedirs('/tmp/.u2net', exist_ok=True)
-
-# Log ayarı
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-# Flask uygulaması
 app = Flask(__name__)
-init_db(app)
 
+# Paylaşılan veri
+shared_data = {
+    "users": [
+        {"id": 1, "name": "Ahmet", "score": 85},
+        {"id": 2, "name": "Mehmet", "score": 92}
+    ],
+    "models": ["model1", "model2", "model3"]
+}
 
 # Ana sayfa
 @app.route('/')
 def home():
-    return "Anime İşleme API - Çalışıyor"
-
-# Generate List Endpoint'leri
-@app.route('/generate_list', methods=['GET'])
-def handle_get_generate_list():
-    try:
-        db = get_db()
-        items = get_generate_list(db)
-        return jsonify({
-            "data": {
-                "generate_list": [dict(item) for item in items]
-            },
-            "success": True
-        })
-    except Exception as e:
-        logger.error(f"Hata: {str(e)}")
-        return jsonify({
-            "data": None,
-            "success": False,
-            "error": str(e)
-        }), 500
+    return "stablediffusionapi test"
 
+# GET Endpoint'leri
+@app.route('/users', methods=['GET'])
+def get_users():
+    return jsonify({"users": shared_data["users"]})
 
-# Anime dönüştürme endpoint'i
-@app.route('/demo_generate/generate', methods=['POST'])
-def generate():
-    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': 'Geçersiz dosya'}), 400
+@app.route('/random-model', methods=['GET'])
+def random_model():
+    return jsonify({"selected_model": random.choice(shared_data["models"])})
 
-        img_io = generate_anime(file.read())
-        return send_file(img_io, mimetype='image/png')
-        
-    except Exception as e:
-        logger.error(f"Hata: {str(e)}")
-        return jsonify({'error': 'İşlem sırasında hata oluştu'}), 500
+@app.route('/user/<int:user_id>', methods=['GET'])
+def get_user(user_id):
+    user = next((u for u in shared_data["users"] if u["id"] == user_id), None)
+    return jsonify(user) if user else (jsonify({"error": "User not found"}), 404)
 
-# Arkaplan kaldırma endpoint'i
-@app.route('/demo_generate/remove-bg', methods=['POST'])
-def bg_remove():
+# POST Endpoint'leri
+@app.route('/predict', methods=['POST'])
+def predict():
     try:
-        if 'image' not in request.files:
-            return jsonify({'error': 'Resim yüklenmedi'}), 400
-            
-        file = request.files['image']
-        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)
-        return send_file(img_io, mimetype='image/png')
-        
+        if not request.json:
+            return jsonify({"error": "No data received"}), 400
+        return jsonify({"probability_up": np.random.rand()})
     except Exception as e:
-        logger.error(f"Arkaplan kaldırma hatası: {str(e)}")
-        return jsonify({'error': str(e)}), 500
+        return jsonify({"error": str(e)}), 500
 
-# Halftone efekti endpoint'i
-@app.route('/demo_generate/halftone', methods=['POST'])
-def halftone_route():
+@app.route('/add-user', methods=['POST'])
+def add_user():
     try:
-        if 'image' not in request.files:
-            return jsonify({'error': 'Resim yüklenmedi'}), 400
+        if not request.json or "name" not in request.json:
+            return jsonify({"error": "Name is required"}), 400
             
-        file = request.files['image']
-        if file.filename == '':
-            return jsonify({'error': 'Boş dosya'}), 400
-        
-        dot_size = int(request.form.get('dot_size', 10))
-        dot_size = max(5, min(20, dot_size))
-        
-        output = apply_halftone(file.read(), dot_size)
-        
-        img_io = io.BytesIO()
-        output.save(img_io, 'PNG')
-        img_io.seek(0)
-        return send_file(img_io, mimetype='image/png')
-        
+        new_user = {
+            "id": len(shared_data["users"]) + 1,
+            "name": request.json["name"],
+            "score": request.json.get("score", 0)
+        }
+        shared_data["users"].append(new_user)
+        return jsonify({"message": "User added", "user": new_user}), 201
     except Exception as e:
-        logger.error(f"Halftone hatası: {str(e)}")
-        return jsonify({'error': str(e)}), 500
-
-# Veritabanı işlemleri
-@app.route('/users', methods=['GET'])
-def get_users_route():
-    from get_methods import get_users
-    return get_users()
-
-@app.route('/user/<string:platform>/<string:udid>', methods=['GET'])
-def get_user_route(udid, platform):
-    from get_methods import get_user
-    return get_user(udid, platform)
-
-@app.route('/add-user', methods=['POST'])
-def add_user_route():
-    from post_methods import add_user
-    return add_user()
-
-
-# Kapatma
-@app.teardown_appcontext
-def shutdown_session(exception=None):
-    close_db()
+        return jsonify({"error": str(e)}), 500
 
-    
-app.run(host='0.0.0.0', port=7860, debug=True)
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)