Update app.py

app.py

@@ -36,7 +36,7 @@ def fetch_logs(file_path):
         conn = sqlite3.connect(file_path)
         cursor = conn.cursor()
 
-        cursor.execute("SELECT * FROM logs;")  # Adjust this query based on your table schema
+        cursor.execute("SELECT * FROM logs2;")  # Adjust this query based on your table schema
         logs = cursor.fetchall()
 
         for log in logs:
@@ -70,9 +70,48 @@ def init_db(file='logs.db'):
     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_request(prompt, is_negative, steps, cfg_scale, sampler, seed, strength, use_dev, 
+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()