Update app.py

app.py

@@ -1,16 +1,51 @@
+import os
+import requests
 import gradio as gr
 from gradio_client import Client
 import matplotlib.pyplot as plt
-import numpy as np
 import io
 from PIL import Image
 import base64
 
-#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")
 
+def get_dynamic_endpoint():
+    """
+    Fetch the dynamic endpoint using the Hugging Face API.
+
+    Returns:
+        str: The current dynamic endpoint.
+    """
+    api_url = "https://api.huggingface.co/space/duchaba/friendly-text-moderation"
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+
+    response = requests.get(api_url, headers=headers)
+    response.raise_for_status()  # Raise an error for bad status codes
+
+    # Extract the endpoint from the response
+    data = response.json()
+    endpoint = data.get("url")
+    return endpoint
+
+# Fetch the dynamic endpoint
+dynamic_endpoint = get_dynamic_endpoint()
+
+# Initialize the client with the dynamic endpoint
+client = Client(dynamic_endpoint, hf_token=HF_TOKEN)
 
 def moderate_text(text, safer_value):
+    """
+    Moderates the given text using the Hugging Face API and returns the result and moderation pie chart.
+
+    Args:
+        text (str): The text to be moderated.
+        safer_value (float): The safer value for text moderation.
+
+    Returns:
+        result (dict): The moderation result.
+        img (PIL.Image): The moderation pie chart.
+    """
     result = client.predict(
         text,
         safer_value,
@@ -35,7 +70,10 @@ demo = gr.Interface(
         gr.Textbox(label="Enter Text:", placeholder="Type your text here...", lines=5),
         gr.Slider(minimum=0.005, maximum=0.1, value=0.005, label="Personalize Safer Value: (larger value is less safe)")
     ],
-    outputs=[gr.Textbox(label="Moderated Text:", lines=5), gr.Image(type="pil", label="Moderation Pie Chart")],
+    outputs=[
+        gr.Textbox(label="Moderated Text:", lines=5),
+        gr.Image(type="pil", label="Moderation Pie Chart")
+    ],
     title="Friendly Text Moderator",
     description="Enter text to be moderated and adjust the safer value to see how it affects the moderation.",
     theme="compact"
@@ -69,4 +107,4 @@ body {
 demo.css = custom_css
 
 # Launch the app
-demo.launch()
+demo.launch()