Update app.py

app.py

@@ -9,6 +9,7 @@ import base64
 import os
 import json
 import logging
+import time
 
 import gradio as gr
 import httpx
@@ -73,25 +74,13 @@ def moderate_content(text_input, image_input, video_input, audio_input, language
         return "Please provide at least one input (text, image, video, or audio).", None
 
     logging.info("Preparing payload for moderation API...")
-    payload = {
-        "model": "nai-moderation-latest" # This is the model name expected by our API
-    }
-
-    if text_input:
-        payload["input"] = text_input
-    
-    image_b64 = file_to_base64(image_input)
-    if image_b64:
-        payload["image"] = image_b64
-
-    video_b64 = file_to_base64(video_input)
-    if video_b64:
-        payload["video"] = video_b64
-
-    audio_b64 = file_to_base64(audio_input)
-    if audio_b64:
+    payload = { "model": "nai-moderation-latest" }
+    if text_input: payload["input"] = text_input
+    if image_b64 := file_to_base64(image_input): payload["image"] = image_b64
+    if video_b64 := file_to_base64(video_input): payload["video"] = video_b64
+    if audio_b64 := file_to_base64(audio_input):
         payload["voice"] = audio_b64
-        language_code = SORTED_LANGUAGES.get(language_full_name, "en") # Default to 'en'
+        language_code = SORTED_LANGUAGES.get(language_full_name, "en")
         payload["language"] = language_code
         logging.info(f"Audio detected. Using language: {language_full_name} ({language_code})")
 
@@ -99,14 +88,19 @@ def moderate_content(text_input, image_input, video_input, audio_input, language
     
     summary_output = "An error occurred. Please check the logs."
     full_response_output = {}
+    latency_ms = None
 
     try:
         with httpx.Client(timeout=180.0) as client:
+            start_time = time.monotonic()
             response = client.post(MODERATION_ENDPOINT, json=payload)
-            response.raise_for_status() # Raises HTTPStatusError for 4xx/5xx responses
+            latency_ms = (time.monotonic() - start_time) * 1000
+            logging.info(f"API response received in {latency_ms:.2f} ms with status code {response.status_code}")
+            
+            response.raise_for_status()
             
             data = response.json()
-            full_response_output = data
+            full_response_output = data # <-- MODIFIED: Assign raw data, without adding latency
             
             if not data.get("results"):
                 summary_output = "API returned an empty result. This might happen if media processing fails (e.g., a video with no frames)."
@@ -121,6 +115,8 @@ def moderate_content(text_input, image_input, video_input, audio_input, language
             categories_str = ", ".join(flagged_categories) if flagged_categories else "None"
 
             summary_output = f"""
+            **API Latency:** {latency_ms:.2f} ms
+            ---
             **Moderation Status:** {status}
             ---
             **Reason:** {reason}
@@ -130,27 +126,29 @@ def moderate_content(text_input, image_input, video_input, audio_input, language
             **Transcribed Text (from audio):**
             {transcribed}
             """
-            logging.info("Successfully received and parsed moderation response.")
+            logging.info("Successfully parsed moderation response.")
 
-    # --- MODIFIED: Enhanced Error Handling ---
     except httpx.HTTPStatusError as e:
-        # Catches errors returned by the backend API (e.g., 422, 500)
         user_message = "The moderation service returned an error."
         error_details = ""
+        latency_str = f"**API Latency:** {latency_ms:.2f} ms" if latency_ms is not None else ""
+        
         try:
-            # Try to parse the JSON error response from the server
             error_json = e.response.json()
             detail = error_json.get("detail", "No specific error detail provided.")
             error_details = f"**Reason:** {detail}"
+            # <-- MODIFIED: Latency removed from this dictionary
             full_response_output = {"error": "Backend API Error", "status_code": e.response.status_code, "details": error_json}
         except (json.JSONDecodeError, AttributeError):
-            # Fallback for non-JSON or unexpected error formats
             error_details = f"**Raw Server Response:**\n```\n{e.response.text}\n```"
+            # <-- MODIFIED: Latency removed from this dictionary
             full_response_output = {"error": "Backend API Error", "status_code": e.response.status_code, "details": e.response.text}
 
         summary_output = f"""
         **🚫 Error from Moderation Service (HTTP {e.response.status_code})**
         ---
+        {latency_str}
+
         {user_message}
         
         {error_details}
@@ -158,19 +156,21 @@ def moderate_content(text_input, image_input, video_input, audio_input, language
         logging.error(f"HTTP Status Error: {e.response.status_code} - Response: {e.response.text}")
     
     except httpx.RequestError as e:
-        # Catches network errors (e.g., server is down, DNS issues)
+        if latency_ms is None:
+            latency_ms = (time.monotonic() - start_time) * 1000 if 'start_time' in locals() else 0
+        
         summary_output = f"""
         **🔌 Connection Error**
         ---
-        Could not connect to the API server at `{API_BASE_URL}`.
+        Could not connect to the API server at `{API_BASE_URL}`. The request failed after {latency_ms:.0f} ms.
         
         Please ensure the backend server is running and the URL is configured correctly in your `.env` file.
         """
+        # <-- MODIFIED: Latency removed from this dictionary
         full_response_output = {"error": "Connection Error", "url": API_BASE_URL, "details": str(e)}
         logging.error(f"Request Error: Could not connect to {API_BASE_URL}. Details: {e}")
 
     except Exception as e:
-        # A catch-all for any other unexpected errors in this Gradio script
         summary_output = f"""
         **💥 An Unexpected Application Error Occurred**
         ---