Update app.py

app.py

@@ -1,6 +1,7 @@
-from deliverable2 import URLValidator
 import os
 import json
+from deliverable2 import URLValidator
+from transformers import pipeline
 
 # Retrieve API key from environment variables
 SERPAPI_API_KEY = os.getenv("SERPAPI_API_KEY")
@@ -45,12 +46,38 @@ urls = [
 # Process queries and URLs separately but use them together
 results = []
 for query, url in zip(queries, urls):
-    result = validator.evaluate_url(url, query)  # Ensure this method is available in the class
-    results.append({
-        "Query": query,
-        "URL": url,
-        "Result": result
-    })
+    try:
+        # Get content from the URL
+        content = validator.fetch_page_content(url)
+        if content:
+            # Get the fake news score and relevance score
+            fake_news_score = validator.get_domain_trust_huggingface(content)  # Fix for long text
+            relevance_score = validator.get_content_relevance(query, content)
+
+            # Store the results
+            results.append({
+                "Query": query,
+                "URL": url,
+                "Trust": fake_news_score,
+                "Relevance": relevance_score
+            })
+        else:
+            results.append({
+                "Query": query,
+                "URL": url,
+                "Trust": "Error fetching content",
+                "Relevance": 0.0
+            })
+
+    except Exception as e:
+        # Handle any unexpected errors
+        results.append({
+            "Query": query,
+            "URL": url,
+            "Trust": "Error",
+            "Relevance": "Error",
+            "Error": str(e)
+        })
 
 # Print formatted output
 print(json.dumps(results, indent=2))