Update app.py

app.py

@@ -7,21 +7,28 @@ validator = URLValidator()
 def validate_url(user_query, url_to_check):
     result = validator.rate_url_validity(user_query, url_to_check)
 
+    # ✅ Ensure result is always a dictionary
+    if not isinstance(result, dict):
+        return {"error": "Unexpected error occurred, please try again."}
+
     # ✅ Check if an error occurred before accessing "raw_score"
     if "Validation Error" in result:
-        return f"⚠️ {result['Validation Error']}"
+        return {"error": result["Validation Error"]}
 
-    # ✅ Access safely without crashing
+    # ✅ Ensure "raw_score" exists in result
+    raw_score = result.get("raw_score", {})
+    
     return {
-        "Domain Trust": result.get("raw_score", {}).get("Domain Trust", "N/A"),
-        "Content Relevance": result.get("raw_score", {}).get("Content Relevance", "N/A"),
-        "Fact-Check Score": result.get("raw_score", {}).get("Fact-Check Score", "N/A"),
-        "Bias Score": result.get("raw_score", {}).get("Bias Score", "N/A"),
-        "Final Validity Score": result.get("raw_score", {}).get("Final Validity Score", "N/A"),
+        "Domain Trust": raw_score.get("Domain Trust", "N/A"),
+        "Content Relevance": raw_score.get("Content Relevance", "N/A"),
+        "Fact-Check Score": raw_score.get("Fact-Check Score", "N/A"),
+        "Bias Score": raw_score.get("Bias Score", "N/A"),
+        "Final Validity Score": raw_score.get("Final Validity Score", "N/A"),
         "Star Rating": result.get("stars", {}).get("icon", "N/A"),
         "Explanation": result.get("explanation", "No explanation available.")
     }
 
+
 # UI Components
 query_options = [
     "How does climate change impact global weather?",