Update src/gradio_server.py

src/gradio_server.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import json
 import os
 import sys
 from fastapi import FastAPI
@@ -96,7 +97,10 @@ def gradio_detect_radicalization(user_text: str):
 
 def gradio_generate_policy_enforcement(user_text: str, violation_context: str):
     # violation_context needs to be provided in a valid JSON format
-    context_dict = eval(violation_context)  # Replace eval with json.loads for safer parsing if it's JSON
+    try:
+        context_dict = json.loads(violation_context)  # Parse violation_context as JSON
+    except json.JSONDecodeError:
+        return {"error": "Invalid JSON format for violation_context"}
     request = PolicyEnforcementRequest(user_text=user_text, violation_context=context_dict)
     response = generate_policy_enforcement(request)
     return response.values
@@ -113,7 +117,7 @@ iface = gr.Interface(
 # Second interface for policy enforcement
 iface2 = gr.Interface(
     fn=gradio_generate_policy_enforcement,  # Function to generate policy enforcement
-    inputs=["text", "json"],  # Two text inputs, one for user text, one for violation context
+    inputs=["text", gr.JSON()],  # Two text inputs, one for user text, one for violation context
     outputs="json",  # Return JSON output
     title="Policy Enforcement Decision",
     description="Enter user text and context to generate a policy enforcement decision."