Update app.py

app.py

@@ -23,7 +23,7 @@ load_dotenv()
 API_BASE_URL = os.environ.get("API_BASE_URL", "")
 MODERATION_ENDPOINT = f"{API_BASE_URL}/v3/moderations"
 
-# --- NEW: Full list of Whisper V3 supported languages ---
+# --- Full list of Whisper V3 supported languages ---
 # Mapping user-friendly names to ISO 639-1 codes
 WHISPER_LANGUAGES = {
     "English": "en", "Chinese": "zh", "German": "de", "Spanish": "es", "Russian": "ru",
@@ -80,7 +80,6 @@ def moderate_content(text_input, image_input, video_input, audio_input, language
     if text_input:
         payload["input"] = text_input
     
-    # Gradio provides file paths; we need to convert them to base64
     image_b64 = file_to_base64(image_input)
     if image_b64:
         payload["image"] = image_b64
@@ -92,22 +91,19 @@ def moderate_content(text_input, image_input, video_input, audio_input, language
     audio_b64 = file_to_base64(audio_input)
     if audio_b64:
         payload["voice"] = audio_b64
-        # --- NEW: Add selected language to the payload ---
-        language_code = SORTED_LANGUAGES.get(language_full_name, "en") # Default to 'en' if not found
+        language_code = SORTED_LANGUAGES.get(language_full_name, "en") # Default to 'en'
         payload["language"] = language_code
         logging.info(f"Audio detected. Using language: {language_full_name} ({language_code})")
 
-
     logging.info(f"Sending request to {MODERATION_ENDPOINT} with inputs: {list(payload.keys())}")
     
     summary_output = "An error occurred. Please check the logs."
     full_response_output = {}
 
     try:
-        # Using a synchronous client is simpler for this Gradio function
         with httpx.Client(timeout=180.0) as client:
             response = client.post(MODERATION_ENDPOINT, json=payload)
-            response.raise_for_status() # Raises an exception for 4xx/5xx errors
+            response.raise_for_status() # Raises HTTPStatusError for 4xx/5xx responses
             
             data = response.json()
             full_response_output = data
@@ -116,15 +112,11 @@ def moderate_content(text_input, image_input, video_input, audio_input, language
                 summary_output = "API returned an empty result. This might happen if media processing fails (e.g., a video with no frames)."
                 return summary_output, full_response_output
 
-            # The v3 endpoint returns a single aggregated result
             result = data["results"][0]
             
-            # Format a nice, human-readable summary
             status = "🚨 FLAGGED 🚨" if result["flagged"] else "✅ SAFE ✅"
             reason = result.get("reason") or "N/A"
             transcribed = result.get("transcribed_text") or "N/A"
-
-            # Create a clean list of flagged categories
             flagged_categories = [cat for cat, flagged in result.get("categories", {}).items() if flagged]
             categories_str = ", ".join(flagged_categories) if flagged_categories else "None"
 
@@ -140,16 +132,55 @@ def moderate_content(text_input, image_input, video_input, audio_input, language
             """
             logging.info("Successfully received and parsed moderation response.")
 
+    # --- MODIFIED: Enhanced Error Handling ---
     except httpx.HTTPStatusError as e:
-        error_details = e.response.text
-        summary_output = f"HTTP Error: {e.response.status_code}\n\nCould not connect to the moderation service or the service returned an error.\n\nDetails:\n{error_details}"
-        logging.error(f"HTTP Status Error: {error_details}")
+        # Catches errors returned by the backend API (e.g., 422, 500)
+        user_message = "The moderation service returned an error."
+        error_details = ""
+        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}"
+            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```"
+            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})**
+        ---
+        {user_message}
+        
+        {error_details}
+        """
+        logging.error(f"HTTP Status Error: {e.response.status_code} - Response: {e.response.text}")
+    
     except httpx.RequestError as e:
-        summary_output = f"Request Error: Could not connect to the API server at {API_BASE_URL}.\nPlease ensure the server is running and the URL is correct."
-        logging.error(f"Request Error: {e}")
+        # Catches network errors (e.g., server is down, DNS issues)
+        summary_output = f"""
+        **🔌 Connection Error**
+        ---
+        Could not connect to the API server at `{API_BASE_URL}`.
+        
+        Please ensure the backend server is running and the URL is configured correctly in your `.env` file.
+        """
+        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:
-        summary_output = f"An unexpected error occurred: {str(e)}"
-        logging.error(f"Unexpected Error: {e}", exc_info=True)
+        # A catch-all for any other unexpected errors in this Gradio script
+        summary_output = f"""
+        **💥 An Unexpected Application Error Occurred**
+        ---
+        An error happened within the Gradio application itself. 
+        Please check the application logs for more details.
+        
+        **Error Type:** `{type(e).__name__}`
+        """
+        full_response_output = {"error": "Gradio App Internal Error", "type": type(e).__name__, "details": str(e)}
+        logging.error(f"Unexpected Error in Gradio App: {e}", exc_info=True)
         
     return summary_output, full_response_output
 
@@ -171,7 +202,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css="footer {display: none !important}")
             video_input = gr.Video(label="Video Input")
             audio_input = gr.Audio(label="Voice/Audio Input", type="filepath")
             
-            # --- NEW: Language selection dropdown ---
             language_input = gr.Dropdown(
                 label="Audio Language (if providing audio)",
                 choices=list(SORTED_LANGUAGES.keys()),
@@ -188,7 +218,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css="footer {display: none !important}")
 
     submit_button.click(
         fn=moderate_content,
-        # --- UPDATED: Add language_input to the list ---
         inputs=[text_input, image_input, video_input, audio_input, language_input],
         outputs=[result_output, full_response_output]
     )
@@ -198,13 +227,11 @@ with gr.Blocks(theme=gr.themes.Soft(), css="footer {display: none !important}")
             ["This is a test of the system with safe text.", None, None, None, "English"],
             ["I am going to kill the process on my computer.", None, None, None, "English"],
         ],
-        # --- UPDATED: Add language_input to the list ---
         inputs=[text_input, image_input, video_input, audio_input, language_input],
         outputs=[result_output, full_response_output],
         fn=moderate_content
     )
 
-
 if __name__ == "__main__":
     logging.info(f"Connecting to API server at: {API_BASE_URL}")
     if API_BASE_URL == "http://127.0.0.1:8000":