Update app.py
Browse files
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 |
-
```
|
|
|
|
| 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:
|