yusufgundogdu commited on
Commit
8aa61de
·
verified ·
1 Parent(s): a240481

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -10
app.py CHANGED
@@ -5,7 +5,7 @@ import logging
5
  from pathlib import Path
6
  from flask import Flask, request, send_file, jsonify, send_from_directory
7
  from PIL import Image
8
- from database import init_db, close_db, get_db, init_generate_list_table, get_generate_list, add_to_generate_list
9
 
10
  # Proje yolu ayarı
11
  sys.path.insert(0, str(Path(__file__).parent))
@@ -171,20 +171,33 @@ if __name__ == '__main__':
171
  # Default modelleri ekle
172
  with app.app_context():
173
  db = get_db()
174
- init_generate_list_table(db)
175
 
176
- # Default modelleri ekle (sadece tablo boşsa)
177
  cursor = db.cursor()
 
 
 
 
 
 
 
 
 
 
178
  cursor.execute('SELECT COUNT(*) FROM generate_list')
179
  if cursor.fetchone()[0] == 0:
180
  default_models = [
181
- ("REALISTIC", "assets/images/0.jpg", 0),
182
- ("DRAWING", "assets/images/1.jpg", 10),
183
- ("TOY FIGURE", "assets/images/2.jpg", 10),
184
- ("3D", "assets/images/3.jpg", 40),
185
- ("ANIMATION", "assets/images/4.jpg", 50)
186
  ]
187
- for title, image_url, contour in default_models:
188
- add_to_generate_list(db, title, image_url, contour)
 
 
 
 
189
 
190
  app.run(host='0.0.0.0', port=7860, debug=True)
 
5
  from pathlib import Path
6
  from flask import Flask, request, send_file, jsonify, send_from_directory
7
  from PIL import Image
8
+ from database import init_db, close_db, get_db, get_generate_list, add_to_generate_list
9
 
10
  # Proje yolu ayarı
11
  sys.path.insert(0, str(Path(__file__).parent))
 
171
  # Default modelleri ekle
172
  with app.app_context():
173
  db = get_db()
 
174
 
175
+ # Tablo yoksa oluştur ve default verileri ekle
176
  cursor = db.cursor()
177
+ cursor.execute('''
178
+ CREATE TABLE IF NOT EXISTS generate_list (
179
+ id INTEGER PRIMARY KEY AUTOINCREMENT,
180
+ title TEXT NOT NULL,
181
+ image_url TEXT NOT NULL,
182
+ contour INTEGER NOT NULL,
183
+ item_order INTEGER NOT NULL DEFAULT 0
184
+ )
185
+ ''')
186
+
187
  cursor.execute('SELECT COUNT(*) FROM generate_list')
188
  if cursor.fetchone()[0] == 0:
189
  default_models = [
190
+ ("REALISTIC", "assets/images/0.jpg", 0, 0),
191
+ ("DRAWING", "assets/images/1.jpg", 10, 1),
192
+ ("TOY FIGURE", "assets/images/2.jpg", 10, 2),
193
+ ("3D", "assets/images/3.jpg", 40, 3),
194
+ ("ANIMATION", "assets/images/4.jpg", 50, 4)
195
  ]
196
+ cursor.executemany('''
197
+ INSERT INTO generate_list (title, image_url, contour, item_order)
198
+ VALUES (?, ?, ?, ?)
199
+ ''', default_models)
200
+
201
+ db.commit()
202
 
203
  app.run(host='0.0.0.0', port=7860, debug=True)