Update app.py

app.py

@@ -26,11 +26,15 @@ def detect_harmful_content(text):
     
     results = response.json()
     
+    if not isinstance(results, list):  # Handle unexpected response format
+        return [{"category": "Error", "score": 0, "message": "Invalid response format"}]
+    
     detected = []
     threshold = 0.8  # Set confidence threshold
     for result in results:
-        if result.get('score', 0) >= threshold:
-            detected.append({"category": result.get('label', 'Unknown'), "score": result.get('score', 0)})
+        if isinstance(result, dict) and 'label' in result and 'score' in result:
+            if result['score'] >= threshold:
+                detected.append({"category": result['label'], "score": result['score']})
     
     return detected
 
@@ -50,10 +54,15 @@ def generate_mitigation_response(text, detected_categories):
     response = requests.post(GENERATOR_API_URL, headers=HEADERS, json=payload)
 
     if response.status_code != 200:
-        return "⚠️ Error: Could not generate a response."
+        return "⚠️ Error: Could not generate a response. Please try again later."
 
     generated = response.json()
-    return generated[0].get('generated_text', "No response generated.")
+
+    # Ensure response contains the generated text
+    if isinstance(generated, list) and len(generated) > 0 and 'generated_text' in generated[0]:
+        return generated[0]['generated_text']
+    else:
+        return "⚠️ No valid response generated. Please try again."
 
 # -----------------------------------------------------------
 # STREAMLIT USER INTERFACE
@@ -79,3 +88,4 @@ if st.button("Analyze"):
         st.markdown("### 💡 Mitigation Response")
         mitigation_response = generate_mitigation_response(user_input, detected)
         st.write(mitigation_response)
+