Update app.py

app.py

@@ -10,13 +10,41 @@ from deep_translator import GoogleTranslator
 from gradio_client import Client
 import logging
 from datetime import datetime
+
+import sqlite3
+from datetime import datetime
+
+# Initialize the database
+def init_db(file='logs.db'):
+    conn = sqlite3.connect(file)
+    c = conn.cursor()
+    c.execute('''CREATE TABLE IF NOT EXISTS logs
+                 (timestamp TEXT, message TEXT)''')
+    conn.commit()
+    conn.close()
+
+# Log a request
+def log_request(prompt, is_negative, steps, cfg_scale, sampler, seed, strength, use_dev, enhance_prompt_style, enhance_prompt_option, nemo_enhance_prompt_style, use_mistral_nemo, huggingface_api_key):
+    log_message = f"Request: prompt='{prompt}', is_negative={is_negative}, steps={steps}, cfg_scale={cfg_scale}, "
+    log_message += f"sampler='{sampler}', seed={seed}, strength={strength}, use_dev={use_dev}, "
+    log_message += f"enhance_prompt_style='{enhance_prompt_style}', enhance_prompt_option={enhance_prompt_option}, "
+    log_message += f"nemo_enhance_prompt_style='{nemo_enhance_prompt_style}', use_mistral_nemo={use_mistral_nemo}"
+    if huggingface_api_key:
+        log_message += f"huggingface_api_key='{huggingface_api_key}'"
+    
+    conn = sqlite3.connect('acces_log.log')
+    c = conn.cursor()
+    c.execute("INSERT INTO logs VALUES (?, ?)", (datetime.now().isoformat(), log_message))
+    conn.commit()
+    conn.close()
+    
 # os.makedirs('assets', exist_ok=True)
 if not os.path.exists('icon.png'):
     os.system("wget -O icon.png https://i.pinimg.com/564x/64/49/88/644988c59447eb00286834c2e70fdd6b.jpg")
 API_URL_DEV = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
 API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
 timeout = 100
-
+init_db('acces_log.log')
 
 # Set up logging
 logging.basicConfig(filename='access.log', level=logging.INFO,
@@ -24,7 +52,7 @@ logging.basicConfig(filename='access.log', level=logging.INFO,
 
 
 
-def log_request(prompt, is_negative, steps, cfg_scale, sampler, seed, strength, use_dev, enhance_prompt_style, enhance_prompt_option, nemo_enhance_prompt_style, use_mistral_nemo, huggingface_api_key):
+def log_requestold(prompt, is_negative, steps, cfg_scale, sampler, seed, strength, use_dev, enhance_prompt_style, enhance_prompt_option, nemo_enhance_prompt_style, use_mistral_nemo, huggingface_api_key):
     log_message = f"Request: prompt='{prompt}', is_negative={is_negative}, steps={steps}, cfg_scale={cfg_scale}, "
     log_message += f"sampler='{sampler}', seed={seed}, strength={strength}, use_dev={use_dev}, "
     log_message += f"enhance_prompt_style='{enhance_prompt_style}', enhance_prompt_option={enhance_prompt_option}, "