Update app.py

app.py

@@ -1,10 +1,18 @@
-import gradio as gr
 from deliverable2 import URLValidator
+import os
+import json
 
-# Instantiate the validator
-validator = URLValidator()
+# Retrieve API key from environment variables
+SERPAPI_API_KEY = os.getenv("SERPAPI_API_KEY")
 
-# ✅ Updated queries and URLs
+# 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)
+
+# List of queries
 queries = [
     "How blockchain works",
     "Climate change effects",
@@ -16,9 +24,10 @@ queries = [
     "Electric vehicle benefits",
     "History of the internet",
     "Nutritional benefits of a vegan diet",
-    "Mental health awareness",
+    "Mental health awareness"
 ]
 
+# Corresponding URLs
 urls = [
     "https://www.ibm.com/topics/what-is-blockchain",
     "https://www.nationalgeographic.com/environment/article/climate-change-overview",
@@ -33,31 +42,15 @@ urls = [
     "https://www.who.int/news-room/fact-sheets/detail/mental-health-strengthening-our-response"
 ]
 
-# 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")
-
-    user_query = gr.Dropdown(queries, label="Select a search query:")
-    url_to_check = gr.Dropdown(urls, label="Select a URL to validate:")
-
-    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)
-
-# Launch the app
-app.launch(server_name="0.0.0.0", server_port=7860)
+# Process queries and URLs separately but use them together
+results = []
+for query, url in zip(queries, urls):
+    result = validator.rate_url_validity(query, url)
+    results.append({
+        "Query": query,
+        "URL": url,
+        "Result": result
+    })
+
+# Print formatted output
+print(json.dumps(results, indent=2))