Update app.py

app.py

@@ -7,16 +7,15 @@ from PIL import Image
 import base64
 import os
 import requests
-#client = Client("https://duchaba-friendly-text-moderation.hf.space/--replicas/gffry/")
-#client = Client("https://duchaba-friendly-text-moderation.hf.space/--replicas/6rx2j/")
 
 # Load Hugging Face token from environment variable
-HF_TOKEN = os.getenv("HF_TOKEN") #, "your_default_hf_token")
+HF_TOKEN = os.getenv("HF_TOKEN")
+if not HF_TOKEN:
+    raise ValueError("Hugging Face token not found in environment variables.")
 
 def get_dynamic_endpoint():
     """
     Fetch the dynamic endpoint using the Hugging Face API.
-
     Returns:
         str: The current dynamic endpoint.
     """
@@ -29,11 +28,21 @@ def get_dynamic_endpoint():
     # Extract the endpoint from the response
     data = response.json()
     endpoint = data.get("url")
+    if not endpoint:
+        raise ValueError("Endpoint URL not found in the response.")
     return endpoint
 
-
-
 def moderate_text(text, safer_value):
+    """
+    Moderate the given text using the Hugging Face API.
+    
+    Args:
+        text (str): The text to be moderated.
+        safer_value (float): The safer value to be used for moderation.
+    
+    Returns:
+        tuple: A tuple containing the moderation result and the generated image.
+    """
     # Fetch the dynamic endpoint
     dynamic_endpoint = get_dynamic_endpoint()
 
@@ -45,8 +54,10 @@ def moderate_text(text, safer_value):
         api_name="/censor_me"
     )
 
-    # Example structure of the result
-    base64_data = result.get('plot', '').split(',')[1]
+    # 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)
@@ -97,4 +108,4 @@ body {
 demo.css = custom_css
 
 # Launch the app
-demo.launch()
+demo.launch()