detect potential commercial missuse

app.py

@@ -79,7 +79,8 @@ def push_to_hf_dataset():
             "source_text": [item["source_text"] for item in translations_buffer],
             "translated_text": [item["translated_text"] for item in translations_buffer],
             "model_used": [item["model_used"] for item in translations_buffer],
-            "timestamp": [item["timestamp"] for item in translations_buffer]
+            "timestamp": [item["timestamp"] for item in translations_buffer],
+            "user_id": [item["user_id"] for item in translations_buffer]  # Include user ID
         })
         
         # Try to load existing dataset
@@ -135,7 +136,7 @@ def translate_ultra_supreme(text, model_name):
     translation = translator(text)[0]['translation_text']
     return translation
 
-def translate_text(text, model_choice):
+def translate_text(text, model_choice, request: gr.Request):
     """ Main translation function """
     global translations_buffer, last_push_time
     
@@ -143,6 +144,11 @@ def translate_text(text, model_choice):
     if not text or text.strip() == "":
         return "Please enter text to translate."
     
+    # Get the user ID (if logged in): to detect potential commercial missuse
+    user_id = "anonymous"
+    if request and hasattr(request, "username") and request.username:
+        user_id = request.username
+    
     # Perform translation
     if model_choice in ["Terjman-Nano-v2", "Terjman-Large-v2"]:
         translation = translate_nano_large(text, model_choice)
@@ -156,7 +162,8 @@ def translate_text(text, model_choice):
         "source_text": text,
         "translated_text": translation,
         "model_used": model_choice,
-        "timestamp": datetime.datetime.now().isoformat()
+        "timestamp": datetime.datetime.now().isoformat(),
+        "user_id": user_id  # Add the user ID to the dataset
     })
     
     # Check if it's time to push to HF
@@ -183,13 +190,13 @@ def gradio_app():
         translate_button = gr.Button("Translate")
         
         # Link input and output
-        def translate_and_update_status(text, model):
-            translation = translate_text(text, model)
+        def translate_and_update_status(text, model, request: gr.Request):
+            translation = translate_text(text, model, request)
             return translation
         
         translate_button.click(
             fn=translate_and_update_status,
-            inputs=[input_text, model_choice],
+            inputs=[input_text, model_choice, gr.Request()],
             outputs=[output_text]
         )