Update app.py

app.py

@@ -15,59 +15,6 @@ def is_token_configured():
     return "✅ API token is configured"
 
 
-def check_safety(input_text, uploaded_image):
-    if not input_text.strip() and uploaded_image is None:
-        return "⚠️ Please enter text or upload an image to check."
-    
-    payload = {}
-    
-    if input_text.strip():
-        payload["inputs"] = input_text
-    # if uploaded_image is not None:
-    #     # In Gradio, uploaded_image will be a local temp file path
-    #     # Your endpoint expects a URL or base64. Here, we send as base64.
-    #     import base64
-        
-    #     with open(uploaded_image, "rb") as img_file:
-    #         img_bytes = img_file.read()
-    #         img_base64 = base64.b64encode(img_bytes).decode('utf-8')
-        
-    #     payload["image"] = img_base64  # Assume your backend can accept image this way
-
-    headers = {
-        "Content-Type": "application/json",
-        "Authorization": f"Bearer {HF_API_TOKEN}"
-    }
-    
-    try:
-        response = requests.post(ENDPOINT_URL, json=payload, headers=headers, timeout=30)
-        
-        if response.status_code == 200:
-            result = response.json()
-            
-            is_safe = result.get("is_safe", False)
-            safety_result = result.get("safety_result", {})
-
-            safety = safety_result.get("Safety", "Unknown")
-            score = safety_result.get("Score", "")
-            categories = safety_result.get("Unsafe Categories", "")
-
-            if is_safe:
-                return f"✅ Safe\n\nSafety: {safety}\nScore: {score}\nUnsafe Categories: {categories}"
-            else:
-                return f"❌ Unsafe\n\nSafety: {safety}\nScore: {score}\nUnsafe Categories: {categories}"
-        
-        else:
-            return f"❗ Error: Request failed with status code {response.status_code}.\nDetails: {response.text}"
-    
-    except requests.exceptions.Timeout:
-        return "❗ Error: Request timed out."
-    
-    except requests.exceptions.ConnectionError:
-        return "❗ Error: Failed to connect to the endpoint."
-    
-    except Exception as e:
-        return f"❗ Error: {str(e)}"
 with gr.Blocks(title="Safety Content Classifier", css="footer {display: none !important}") as demo:
     gr.Markdown(f"# Safety Content Classifier")
     gr.Markdown(f"## Connected to external safety model endpoint")
@@ -93,10 +40,7 @@ with gr.Blocks(title="Safety Content Classifier", css="footer {display: none !im
                 placeholder="Type here...",
                 lines=5
             )
-            uploaded_image = gr.Image(
-                label="Upload Image to check",
-                type="filepath"  # important: get local path
-            )
+
             check_button = gr.Button("Check Safety", variant="primary")
         
         with gr.Column():