Update app.py

app.py

@@ -1,51 +1,70 @@
-import os
-import spaces
 import urllib.parse
 import requests
 import gradio as gr
 from transformers import pipeline
 
+# 🟢 API Templates للحصول على بيانات ويكيبيديا
 SEARCH_TEMPLATE = "https://ar.wikipedia.org/w/api.php?action=opensearch&search=%s&limit=1&namespace=0&format=json"
 CONTENT_TEMPLATE = "https://ar.wikipedia.org/w/api.php?format=json&action=query&prop=extracts&exintro&explaintext&redirects=1&titles=%s"
 
-summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
+# 🔥 استخدم نموذج يدعم اللغة العربية مثل "facebook/bart-large-cnn" أو أي نموذج متخصص
+summarizer = pipeline("summarization", model="facebook/mbart-large-50", tokenizer="facebook/mbart-large-50")
 
-@spaces.GPU
 def search_wikipedia(query):
-    query = urllib.parse.quote_plus(query)
-    data = requests.get(SEARCH_TEMPLATE % query).json()
-    if data and data[1]:
-        page = urllib.parse.quote_plus(data[1][0])
-        content = requests.get(CONTENT_TEMPLATE % page).json()
-        content = list(content["query"]["pages"].values())[0]["extract"]
-        summary = summarizer(content, max_length=150, min_length=50, do_sample=False)[0]['summary_text']
-        source = data[3][0]
-        return summary, source
-    else:
-        return "No results found.", ""
+    """ البحث في ويكيبيديا العربية وإرجاع ملخص من المقال الأول. """
+    query_encoded = urllib.parse.quote_plus(query)
+    search_response = requests.get(SEARCH_TEMPLATE % query_encoded).json()
+
+    if not search_response or not search_response[1]:  
+        return "❌ لم يتم العثور على نتائج.", ""
+
+    # 🟢 جلب أول نتيجة
+    page_title = search_response[1][0]
+    page_encoded = urllib.parse.quote_plus(page_title)
+    
+    # 🟢 جلب محتوى المقالة
+    content_response = requests.get(CONTENT_TEMPLATE % page_encoded).json()
+    pages = content_response.get("query", {}).get("pages", {})
+    
+    if not pages:
+        return "❌ لم يتم العثور على المحتوى.", ""
+
+    content = list(pages.values())[0].get("extract", "")
+
+    if not content:
+        return "❌ المقالة لا تحتوي على معلومات كافية.", ""
+
+    # 🟢 تحسين التلخيص بناءً على طول المقالة
+    max_length = 200 if len(content) > 1000 else 100
+    min_length = 50 if len(content) > 500 else 30
+
+    summary = summarizer(content, max_length=max_length, min_length=min_length, do_sample=False)[0]['summary_text']
+
+    source = search_response[3][0]  # رابط المقال الأصلي
+    return summary, source
 
 def chatbot_response(message, history):
+    """ دالة التفاعل مع المستخدم """
     summary, source = search_wikipedia(message)
-    response = f"Here's a summary of the top Wikipedia result:\n\n{summary}"
+    response = f"🔹 **ملخص ويكيبيديا:**\n\n{summary}"
     if source:
-        response += f"\n\nSource: {source}"
+        response += f"\n\n🔗 **المصدر:** [اضغط هنا]({source})"
     history.append((message, response))
     return history
 
-demo = gr.Blocks()
-
-with demo:
-    gr.Markdown("# Wikipedia Chatbot")
-    gr.Markdown("This chatbot queries the Wikipedia API and summarizes the top result.")
+# 🔥 واجهة Gradio المحسنة
+with gr.Blocks() as demo:
+    gr.Markdown("# 🤖 بوت ويكيبيديا العربية")
+    gr.Markdown("🔹 هذا البوت يستخدم ويكيبيديا العربية للبحث عن المعلومات وإعطاء ملخص عنها.")
 
     chatbot = gr.Chatbot()
-    msg = gr.Textbox(label="ادخل سؤالك")
-    clear = gr.Button("امسح")
+    msg = gr.Textbox(label="🔍 اكتب سؤالك هنا:")
+    clear = gr.Button("🧹 مسح المحادثة")
 
     msg.submit(chatbot_response, [msg, chatbot], chatbot).then(
-        lambda: gr.update(value=""), None, [msg], queue=False
+        lambda _: "", None, [msg], queue=False  # 🟢 تصحيح التحديث التلقائي لحقل الإدخال
     )
-    clear.click(lambda: None, None, chatbot, queue=False)
+    clear.click(lambda: [], None, chatbot, queue=False)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()