Update app.py

app.py

@@ -8,19 +8,36 @@ from PIL import Image
 from huggingface_hub import InferenceClient
 from deep_translator import GoogleTranslator
 from gradio_client import Client
-
+import logging
+from datetim
 # 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
+e import datetime
+
+# Set up logging
+logging.basicConfig(filename='access.log', level=logging.INFO,
+                    format='%(asctime)s - %(message)s', datefmt='%Y-%m-%d %H:%M:%S')
+
+
+
+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}'"
+    logging.info(log_message)
 
 def check_ubuse(prompt,word_list=["little girl"]):
     for word in word_list:
         if word in prompt:
             print(f"Abuse! prompt {prompt} wiped!")
-            return "Fuck You Bastard!"
+            return "None"
     return prompt
     
 def enhance_prompt(prompt, model="mistralai/Mistral-7B-Instruct-v0.1", style="photo-realistic"):
@@ -97,6 +114,8 @@ def mistral_nemo_call(prompt, API_TOKEN, model="mistralai/Mistral-Nemo-Instruct-
     return response
     
 def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, huggingface_api_key=None, use_dev=False,enhance_prompt_style="generic", enhance_prompt_option=False, nemo_enhance_prompt_style="generic", use_mistral_nemo=False):
+    
+    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)
     # Determine which API URL to use
     api_url = API_URL_DEV if use_dev else API_URL