Update app.py

app.py

@@ -4,18 +4,23 @@ import random
 import time
 from transformers import pipeline
 
+# Load the zero-shot classification pipeline
 classifier = pipeline("zero-shot-classification", model="valhalla/distilbart-mnli-12-3")
-random_topics = ["cats", "space", "chocolate", "Egypt", "Leonardo da Vinci",
-                 "volcanoes", "Tokyo", "honeybees", "quantum physics", "orcas"]
 
-def get_wikipedia_facts(topic):
-    # your existing function here unchanged...
-    # (copy-paste your original function)
+# List of random topics for the "Surprise me" button
+random_topics = [
+    "cats", "space", "chocolate", "Egypt", "Leonardo da Vinci",
+    "volcanoes", "Tokyo", "honeybees", "quantum physics", "orcas"
+]
 
+# Main function: get Wikipedia extract, image, and classify topic
+def get_wikipedia_facts(topic):
     if not topic.strip():
         return "Please enter a topic or use 'Surprise me!'", None, None
 
-    headers = {"User-Agent": "RandomFactApp/3.0 (https://huggingface.co/spaces/yourname) Python requests"}
+    headers = {
+        "User-Agent": "RandomFactApp/3.0 (https://huggingface.co/spaces/yourname) Python requests"
+    }
 
     search_url = "https://en.wikipedia.org/w/api.php"
     search_params = {
@@ -27,6 +32,7 @@ def get_wikipedia_facts(topic):
     }
 
     try:
+        # Step 1: Search Wikipedia
         search_response = requests.get(search_url, params=search_params, headers=headers)
         time.sleep(0.3)
         search_data = search_response.json()
@@ -37,6 +43,7 @@ def get_wikipedia_facts(topic):
 
         best_title = search_hits[0]["title"]
 
+        # Step 2: Get page extract and image
         extract_params = {
             "action": "query",
             "format": "json",
@@ -59,6 +66,7 @@ def get_wikipedia_facts(topic):
         if not extract_text:
             return f"Sorry, no extract found for '{topic}'.", None, None
 
+        # Format extract into short facts
         sentences = [s.strip() for s in extract_text.replace("\n", " ").split(". ") if s.strip()]
         if not sentences:
             return f"Sorry, no facts available for '{topic}'.", None, None
@@ -67,8 +75,10 @@ def get_wikipedia_facts(topic):
         facts = [fact if fact.endswith(".") else fact + "." for fact in facts]
         facts_text = "\n\n".join(f"💡 {fact}" for fact in facts)
 
+        # Get image URL
         image_url = page.get("thumbnail", {}).get("source", None)
 
+        # Zero-shot classification
         labels = ["history", "science", "technology", "art", "geography", "biology", "music", "sports", "politics"]
         classification = classifier(topic, labels)
         top_label = classification["labels"][0]
@@ -81,29 +91,34 @@ def get_wikipedia_facts(topic):
         print("Error:", e)
         return "Oops! Something went wrong while fetching your facts.", None, None
 
+# Surprise topic function
 def surprise_topic(_):
     topic = random.choice(random_topics)
     return get_wikipedia_facts(topic)
 
+# Gradio UI layout
 with gr.Blocks() as demo:
 
-    # Background box with soft clouds image
-    with gr.Box(
-        elem_id="background-box",
-        style={
-            "background-image": "url('https://images.unsplash.com/photo-1506744038136-46273834b3fb?auto=format&fit=crop&w=1470&q=80')",
-            "background-size": "cover",
-            "background-position": "center",
-            "position": "fixed",
-            "top": "0",
-            "left": "0",
-            "width": "100vw",
-            "height": "100vh",
-            "z-index": "-1",
-            "filter": "brightness(0.85)"
+    # Background image injected via HTML (safe for Hugging Face)
+    gr.HTML("""
+    <style>
+        .cloud-bg {
+            background: url('https://images.unsplash.com/photo-1506744038136-46273834b3fb?auto=format&fit=crop&w=1470&q=80') no-repeat center center fixed;
+            background-size: cover;
+            position: fixed;
+            top: 0;
+            left: 0;
+            width: 100vw;
+            height: 100vh;
+            z-index: -1;
+            opacity: 0.8;
         }
-    ):
-        pass
+        .gradio-container {
+            background: transparent !important;
+        }
+    </style>
+    <div class="cloud-bg"></div>
+    """)
 
     gr.Markdown("""
     # 🌍 Smart Wikipedia Fact Finder  
@@ -130,11 +145,12 @@ with gr.Blocks() as demo:
         with gr.Column():
             image_output = gr.Image(label="🖼️ Related Image")
 
+    # Link functions to inputs
     topic_input.submit(get_wikipedia_facts, inputs=topic_input, outputs=[facts_output, image_output, classification_output])
     surprise_button.click(surprise_topic, inputs=None, outputs=[facts_output, image_output, classification_output])
 
+# Run the app
 if __name__ == "__main__":
     demo.launch()
 
-