Update app.py

app.py

@@ -3,6 +3,9 @@ import os
 import sys
 import json
 import random
+import hashlib
+import requests
+from datetime import datetime
 from openai import OpenAI
 
 MODEL = "gpt-4.1-mini"
@@ -32,6 +35,7 @@ def get_env_list(key, default=""):
 DISABLED = get_env_bool("DISABLED", "False")
 OPENAI_API_KEYS = get_env_list("OPENAI_API_KEYS", "")
 NUM_THREADS = int(os.getenv("NUM_THREADS", "4"))
+IP_SALT = os.getenv("IP_SALT", "latamgpt-default-salt-2025")
 
 def exception_handler(exception_type, exception, traceback):
     print(f"{exception_type.__name__}: {exception}")
@@ -39,6 +43,30 @@ def exception_handler(exception_type, exception, traceback):
 sys.excepthook = exception_handler
 sys.tracebacklimit = 0
 
+def get_user_fingerprint(request):
+    real_ip = (
+        request.headers.get('x-forwarded-for', '').split(',')[0].strip() or
+        request.headers.get('x-real-ip', '') or 
+        getattr(request, 'client', {}).get('host', 'unknown')
+    )
+    fingerprint_data = f"{real_ip}:{IP_SALT}"
+    user_fingerprint = hashlib.sha256(fingerprint_data.encode()).hexdigest()[:16]
+    return real_ip, user_fingerprint
+
+def get_country_from_ip(ip):
+    try:
+        response = requests.get(f"http://ip-api.com/json/{ip}", timeout=2)
+        if response.status_code == 200:
+            data = response.json()
+            return {
+                "country": data.get('country', 'Unknown'),
+                "country_code": data.get('countryCode', 'UN'),
+                "region": data.get('regionName', 'Unknown')
+            }
+    except:
+        pass
+    return {"country": "Unknown", "country_code": "UN", "region": "Unknown"}
+
 def predict(inputs, top_p, temperature, chat_counter, chatbot, history, request: gr.Request):
     if not OPENAI_API_KEYS or not OPENAI_API_KEYS[0]:
         yield [(history[i], history[i + 1]) for i in range(0, len(history) - 1, 2)], history, chat_counter, "No API keys configured", gr.update(interactive=True), gr.update(interactive=True)
@@ -47,6 +75,9 @@ def predict(inputs, top_p, temperature, chat_counter, chatbot, history, request:
     api_key = random.choice(OPENAI_API_KEYS)
     client = OpenAI(api_key=api_key)
     
+    session_id = getattr(request, 'session_hash', 'unknown')
+    real_ip, user_fingerprint = get_user_fingerprint(request)
+    geo_info = get_country_from_ip(real_ip)
     headers_dict = {key.decode('utf-8'): value.decode('utf-8') for key, value in request.headers.raw}
     
     messages = []
@@ -84,6 +115,7 @@ def predict(inputs, top_p, temperature, chat_counter, chatbot, history, request:
                 token_counter += 1
                 yield [(history[i], history[i + 1]) for i in range(0, len(history) - 1, 2)], history, chat_counter, "200", gr.update(interactive=False), gr.update(interactive=False)
         
+        # Re-enable inputs after streaming completes
         yield [(history[i], history[i + 1]) for i in range(0, len(history) - 1, 2)], history, chat_counter, "200", gr.update(interactive=True), gr.update(interactive=True)
                 
     except Exception as e:
@@ -91,6 +123,12 @@ def predict(inputs, top_p, temperature, chat_counter, chatbot, history, request:
         yield [(history[i], history[i + 1]) for i in range(0, len(history) - 1, 2)], history, chat_counter, str(e), gr.update(interactive=True), gr.update(interactive=True)
     
     log_data = {
+        "session_id": session_id,
+        "user_fingerprint": user_fingerprint,
+        "conversation_id": f"{session_id}_{datetime.now().strftime('%Y%m%d_%H')}",
+        "country": geo_info["country"],
+        "country_code": geo_info["country_code"],
+        "region": geo_info["region"],
         "chat_counter": chat_counter,
         "model": MODEL,
         "messages": messages,
@@ -98,7 +136,8 @@ def predict(inputs, top_p, temperature, chat_counter, chatbot, history, request:
         "headers": headers_dict,
         "temperature": temperature,
         "top_p": top_p,
-        "token_counter": token_counter
+        "token_counter": token_counter,
+        "timestamp": datetime.now().isoformat()
     }
     print(json.dumps(log_data))