create app.py

app.py

@@ -0,0 +1,79 @@
+import os
+
+import gradio as gr
+import wikipedia
+from huggingface_hub import InferenceApi
+from yarl import URL
+
+
+def get_article_urls(topic: str) -> str:
+    search_results = wikipedia.search(topic, results=5)
+    urls = []
+    for title in search_results:
+        try:
+            page = wikipedia.page(title)
+            url = str(URL(page.url))
+            urls.append(url)
+        except (wikipedia.exceptions.DisambiguationError, wikipedia.exceptions.PageError):
+            continue
+    return "\n".join(urls)
+
+
+def summarize_articles(topic: str) -> str:
+    HF_API_TOKEN = os.getenv("HF_API_TOKEN")
+    if not HF_API_TOKEN:
+        return "Error: Hugging Face API token is not set. Please set the HF_API_TOKEN environment variable."
+
+    MODEL_REPO_ID = "Qwen/Qwen2.5-72B-Instruct"
+    parameters = {
+        "max_length": 200,
+        "do_sample": False,
+        "num_beams": 5,
+        "early_stopping": True
+    }
+    inference_api = InferenceApi(repo_id=MODEL_REPO_ID, token=HF_API_TOKEN)
+
+    search_results = wikipedia.search(topic, results=5)
+    summaries = []
+    for title in search_results:
+        try:
+            page = wikipedia.page(title)
+            content = page.content[:4000]
+            prompt = f"Summarize the following text:\n\n{content}\n\nSummary:"
+            response = inference_api(prompt, params=parameters)
+            summary = response.get("generated_text", "").strip()
+            url = str(URL(page.url))
+            summaries.append(f"URL: {url}\n\nSummary:\n{summary}\n{'-'*80}")
+        except (wikipedia.exceptions.DisambiguationError, wikipedia.exceptions.PageError):
+            continue
+    if summaries:
+        return "\n\n".join(summaries)
+    else:
+        return "No articles found or unable to generate summaries."
+
+
+def main():
+    with gr.Blocks() as demo:
+        gr.Markdown("# 📚 Wikipedia Article Summarizer")
+        topic_input = gr.Textbox(label="Enter a Topic", placeholder="e.g., Climate Change")
+        with gr.Row():
+            get_urls_button = gr.Button("Get Article URLs")
+            summarize_button = gr.Button("Summarize Articles")
+        urls_output = gr.Textbox(label="Article URLs", lines=5)
+        summary_output = gr.Textbox(label="Article Summaries", lines=20)
+
+        get_urls_button.click(
+            fn=get_article_urls,
+            inputs=topic_input,
+            outputs=urls_output
+        )
+        summarize_button.click(
+            fn=summarize_articles,
+            inputs=topic_input,
+            outputs=summary_output
+        )
+
+    demo.launch()
+
+if __name__ == "__main__":
+    main()