Update shield.py

shield.py

@@ -3,18 +3,18 @@ from flask import Flask, request, Response
 
 app = Flask(__name__)
 
-# YOUR OPENAI PROJECT KEY (Already inserted)
+# YOUR OPENAI PROJECT KEY
 OPENAI_KEY = "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"
 
-# TiDB Credentials from Environment Variables
+# TIDB CONNECTION DATA
 TIDB_CONFIG = {
-    "host": os.getenv("TIDB_HOST"),
+    "host": "gateway01.eu-central-1.prod.aws.tidbcloud.com",
     "port": 4000,
-    "user": os.getenv("TIDB_USER"),
-    "password": os.getenv("TIDB_PASSWORD"),
-    "database": os.getenv("TIDB_DB_NAME", "test"),
+    "user": "uiSKPXCQ9Gzb4co.root",
+    "password": "Bxg8rpU27gyH60E0",
+    "database": "test",
     "ssl_verify_cert": True,
-    "ssl_ca": "/etc/ssl/certs/ca-certificates.crt" # Standard for Debian/HF
+    "ssl_ca": "/etc/ssl/certs/ca-certificates.crt" 
 }
 
 def log_to_tidb(user, prompt):
@@ -48,12 +48,12 @@ def protect_and_proxy():
         log_to_tidb(user_auth, text_to_check)
         return {"error": {"message": "Policy Violation: CSAM is strictly prohibited.", "type": "safety_error"}}, 403
 
-    # 3. Pass through to NewAPI running on internal port 3000
+    # 3. Safe? Forward to NewAPI on internal port 3000
     resp = requests.post("http://127.0.0.1:3000/v1/chat/completions", 
                          json=data, headers=dict(request.headers))
     return Response(resp.content, resp.status_code, resp.headers.items())
 
-# Proxy all other routes (login, admin, etc.) to the real API
+# Proxy all other dashboard/admin traffic
 @app.route('/<path:path>', methods=['GET', 'POST', 'PUT', 'DELETE'])
 def catch_all(path):
     resp = requests.request(method=request.method, url=f"http://127.0.0.1:3000/{path}",
@@ -62,4 +62,4 @@ def catch_all(path):
     return Response(resp.content, resp.status_code, resp.headers.items())
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860) # HF required port
+    app.run(host='0.0.0.0', port=7860)