Sada8888 commited on
Commit
3486397
·
verified ·
1 Parent(s): 750e40d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -7
app.py CHANGED
@@ -21,13 +21,12 @@ SPACE_URL = f"https://{SPACE_HOST}"
21
  FORBIDDEN_KEYWORDS = {"sex", "sexy", "porn", "nude", "erotic", "سکس", "سکسی", "پورن", "شهوانی", "برهنه", "لخت"}
22
 
23
  # --- تنظیمات دیتابیس SQLite ---
24
- # در هاگینگ فیس باکت‌ها معمولاً در مسیر /data مونت می‌شوند
25
  DB_DIR = "/data" if os.path.exists("/data") else "."
26
  DB_PATH = os.path.join(DB_DIR, "users.db")
27
  TEMP_DIR = "tmp"
28
  USAGE_LIMIT = 5
29
  ONE_WEEK_SECONDS = 7 * 24 * 60 * 60
30
- ONE_DAY_SECONDS = 24 * 60 * 60 # اضافه شده برای ریست شدن روزانه اعتبارات تصویر
31
 
32
  # متغیرهای مربوط به دیتابیس جدید تصاویر
33
  DB_IMAGE_PATH = os.path.join(DB_DIR, "users_image.db")
@@ -48,7 +47,6 @@ def init_db():
48
  conn.commit()
49
  conn.close()
50
 
51
- # ساخت دیتابیس اختصاصی تصاویر
52
  def init_image_db():
53
  conn = sqlite3.connect(DB_IMAGE_PATH)
54
  c = conn.cursor()
@@ -57,7 +55,6 @@ def init_image_db():
57
  conn.commit()
58
  conn.close()
59
 
60
- # ساخت دیتابیس اختصاصی ویرایش تصاویر
61
  def init_image_edit_db():
62
  conn = sqlite3.connect(DB_IMAGE_EDIT_PATH)
63
  c = conn.cursor()
@@ -268,8 +265,11 @@ def generate_video():
268
 
269
  if gemma_success:
270
  if "```" in english_prompt:
 
 
271
  english_prompt = english_prompt.replace("
272
- ```text", "").replace("```", "").replace("```json", "").strip()
 
273
 
274
  if "پردازش متوقف شد" in english_prompt or "سهمیه سرور موقتاً پر شده" in english_prompt:
275
  print(f"Gemma returned quota error on attempt {attempt + 1}. Retrying...")
@@ -500,7 +500,6 @@ def check_image_credit():
500
  reset_timestamp = 0
501
 
502
  if user_record:
503
- # تغییر از هفته به روزانه
504
  if user_record['week_start'] < (now - ONE_DAY_SECONDS):
505
  c.execute("UPDATE users SET count = 0, week_start = ? WHERE id = ?", (now, user_id))
506
  conn.commit()
@@ -531,7 +530,6 @@ def use_image_credit():
531
 
532
  now = time.time()
533
  if user_record:
534
- # تغییر از هفته به روزانه
535
  if user_record['week_start'] < (now - ONE_DAY_SECONDS):
536
  c.execute("UPDATE users SET count = 1, week_start = ? WHERE id = ?", (now, user_id))
537
  else:
 
21
  FORBIDDEN_KEYWORDS = {"sex", "sexy", "porn", "nude", "erotic", "سکس", "سکسی", "پورن", "شهوانی", "برهنه", "لخت"}
22
 
23
  # --- تنظیمات دیتابیس SQLite ---
 
24
  DB_DIR = "/data" if os.path.exists("/data") else "."
25
  DB_PATH = os.path.join(DB_DIR, "users.db")
26
  TEMP_DIR = "tmp"
27
  USAGE_LIMIT = 5
28
  ONE_WEEK_SECONDS = 7 * 24 * 60 * 60
29
+ ONE_DAY_SECONDS = 24 * 60 * 60
30
 
31
  # متغیرهای مربوط به دیتابیس جدید تصاویر
32
  DB_IMAGE_PATH = os.path.join(DB_DIR, "users_image.db")
 
47
  conn.commit()
48
  conn.close()
49
 
 
50
  def init_image_db():
51
  conn = sqlite3.connect(DB_IMAGE_PATH)
52
  c = conn.cursor()
 
55
  conn.commit()
56
  conn.close()
57
 
 
58
  def init_image_edit_db():
59
  conn = sqlite3.connect(DB_IMAGE_EDIT_PATH)
60
  c = conn.cursor()
 
265
 
266
  if gemma_success:
267
  if "```" in english_prompt:
268
+ english_prompt = english_prompt.replace("```text", "")
269
+ english_prompt = english_prompt.replace("```json", "")
270
  english_prompt = english_prompt.replace("
271
+ ```", "")
272
+ english_prompt = english_prompt.strip()
273
 
274
  if "پردازش متوقف شد" in english_prompt or "سهمیه سرور موقتاً پر شده" in english_prompt:
275
  print(f"Gemma returned quota error on attempt {attempt + 1}. Retrying...")
 
500
  reset_timestamp = 0
501
 
502
  if user_record:
 
503
  if user_record['week_start'] < (now - ONE_DAY_SECONDS):
504
  c.execute("UPDATE users SET count = 0, week_start = ? WHERE id = ?", (now, user_id))
505
  conn.commit()
 
530
 
531
  now = time.time()
532
  if user_record:
 
533
  if user_record['week_start'] < (now - ONE_DAY_SECONDS):
534
  c.execute("UPDATE users SET count = 1, week_start = ? WHERE id = ?", (now, user_id))
535
  else: