Update app.py

app.py

@@ -1,17 +1,8 @@
-import os
-import json
-from deliverable2 import URLValidator
-from transformers import pipeline
+import gradio as gr
+from Deliverable2 import URLValidator
 
-# Retrieve API key from environment variables
-SERPAPI_API_KEY = os.getenv("SERPAPI_API_KEY")
-
-# Ensure API key is provided before initializing
-if not SERPAPI_API_KEY:
-    raise ValueError("SERPAPI_API_KEY is not set. Please configure your environment variables.")
-
-# Initialize the validator correctly
-validator = URLValidator(serpapi_key=SERPAPI_API_KEY)
+# Instantiate the validator
+validator = URLValidator()
 
 # List of queries
 queries = [
@@ -43,41 +34,31 @@ urls = [
     "https://www.who.int/news-room/fact-sheets/detail/mental-health-strengthening-our-response"
 ]
 
-# Process queries and URLs separately but use them together
-results = []
-for query, url in zip(queries, urls):
-    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)
+# Function to validate URL
+def validate_url(user_query, url_to_check):
+    result = validator.rate_url_validity(user_query, url_to_check)
+
+    if "Validation Error" in result:
+        return {"Error": result["Validation Error"]}
+
+    return {
+        "Content Relevance Score": f"{result['raw_score']['Content Relevance']} / 100",
+        "Bias Score": f"{result['raw_score']['Bias Score']} / 100",
+        "Final Validity Score": f"{result['raw_score']['Final Validity Score']} / 100"
+    }
+
+# Gradio UI
+with gr.Blocks() as app:
+    gr.Markdown("# 🌍 URL Credibility Validator")
+    gr.Markdown("### Validate the credibility of any webpage using AI")
 
-            # 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
-            })
+    user_query = gr.Dropdown(queries, label="Select a search query:")
+    url_to_check = gr.Dropdown(urls, label="Select a URL to validate:")
 
-    except Exception as e:
-        # Handle any unexpected errors
-        results.append({
-            "Query": query,
-            "URL": url,
-            "Trust": "Error",
-            "Relevance": "Error",
-            "Error": str(e)
-        })
+    result_output = gr.JSON(label="Validation Results")
+    
+    submit_button = gr.Button("Validate URL")
+    submit_button.click(validate_url, inputs=[user_query, url_to_check], outputs=result_output)
 
-# Print formatted output
-print(json.dumps(results, indent=2))
+# Launch the app
+app.launch(server_name="0.0.0.0", server_port=7860)