Update app.py

app.py

@@ -25,7 +25,7 @@ def get_wikipedia_facts(topic):
     }
 
     try:
-     
+       
         search_response = requests.get(search_url, params=search_params, headers=headers)
         time.sleep(0.3)
         search_response.raise_for_status()
@@ -37,7 +37,7 @@ def get_wikipedia_facts(topic):
 
         best_title = search_hits[0]["title"]
 
-       
+
         extract_params = {
             "action": "query",
             "format": "json",
@@ -61,5 +61,61 @@ def get_wikipedia_facts(topic):
         if not extract_text:
             return f"Sorry, no extract found for '{topic}'.", None, None
 
+      
+        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
+
+        
+        facts = random.sample(sentences, min(3, len(sentences)))
+        facts = [fact if fact.endswith(".") else fact + "." for fact in facts]
+        facts_text = "\n\n".join(f"💡 {fact}" for fact in facts)
+
+        # Get Wikipedia image URL
+        image_url = None
+        if "thumbnail" in page:
+            image_url = page["thumbnail"].get("source")
+
+      
+        candidate_labels = ["history", "science", "technology", "art", "geography", "biology", "music", "sports", "politics"]
+        classification = classifier(topic, candidate_labels)
+        top_label = classification['labels'][0]
+        top_score = classification['scores'][0]
+        classification_text = f"Category: {top_label} (confidence: {top_score:.2f})"
+
+        return facts_text, image_url, classification_text
+
+    except Exception as e:
+        print("Error:", e)
+        return "Oops! Something went wrong while fetching your facts.", None, None
+
+
+random_topics = [
+    "cats", "space", "chocolate", "Egypt", "Leonardo da Vinci",
+    "volcanoes", "Tokyo", "honeybees", "quantum physics", "orcas"
+]
+
+def surprise_topic(_):
+    topic = random.choice(random_topics)
+    return get_wikipedia_facts(topic)
+
+with gr.Blocks() as demo:
+    gr.Markdown("## 🤖 Wikipedia Facts + Image + Topic Classification\nEnter a topic or click 'Surprise me!'")
+
+    with gr.Row():
+        topic_input = gr.Textbox(placeholder="Enter a topic like 'koalas' or 'Eiffel Tower'")
+        surprise_button = gr.Button("🎲 Surprise me!")
+
+    facts_output = gr.Textbox(label="3 Wikipedia Facts")
+    image_output = gr.Image(label="Image from Wikipedia")
+    classification_output = gr.Textbox(label="Topic Classification")
+
+    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])
+
+if __name__ == "__main__":
+    demo.launch()
+
+