Update app.py

app.py

@@ -43,29 +43,35 @@ def moderate_text(text, safer_value):
     Returns:
         tuple: A tuple containing the moderation result and the generated image.
     """
-    # Fetch the dynamic endpoint
-    dynamic_endpoint = get_dynamic_endpoint()
+    try:
+        # Fetch the dynamic endpoint
+        dynamic_endpoint = get_dynamic_endpoint()
 
-    # Initialize the client with the dynamic endpoint
-    client = Client(dynamic_endpoint, hf_token=HF_TOKEN)
-    result = client.predict(
-        text,
-        safer_value,
-        api_name="/censor_me"
-    )
+        # Initialize the client with the dynamic endpoint
+        client = Client(dynamic_endpoint, hf_token=HF_TOKEN)
+        result = client.predict(
+            text,
+            safer_value,
+            api_name="/censor_me"
+        )
 
-    # Ensure the result contains the expected data
-    base64_data = result.get('plot', '').split(',')[1] if 'plot' in result else None
-    if not base64_data:
-        raise ValueError("Expected plot data not found in the result.")
+        # Ensure the result contains the expected data
+        base64_data = result.get('plot', '').split(',')[1] if 'plot' in result else None
+        if not base64_data:
+            raise ValueError("Expected plot data not found in the result.")
 
-    # Decode base64 to bytes
-    img_data = base64.b64decode(base64_data)
+        # Decode base64 to bytes
+        img_data = base64.b64decode(base64_data)
 
-    # Convert bytes to PIL Image
-    img = Image.open(io.BytesIO(img_data))
+        # Convert bytes to PIL Image
+        img = Image.open(io.BytesIO(img_data))
 
-    return result, img
+        return result, img
+    
+    except Exception as e:
+        # Log the error for debugging purposes
+        print(f"Error occurred: {e}")
+        return str(e), None
 
 # Define the Gradio interface
 demo = gr.Interface(