Update app.py

app.py

@@ -1,14 +1,7 @@
-"""
-CyberSec Web Vulnerabilities Assistant
-Gradio app for Hugging Face Spaces - v1.2
-"""
-
 import json
 import gradio as gr
 
-# ─────────────────────────────────────────
-# Load dataset
-# ─────────────────────────────────────────
+# تحميل البيانات
 with open("dataset.json", "r", encoding="utf-8") as f:
     DATA = json.load(f)
 
@@ -57,7 +50,6 @@ ALIASES = {
     "tools": "Additional Tools",
 }
 
-
 def find_category(query: str):
     q = query.strip().lower()
     for alias, cat_name in ALIASES.items():
@@ -73,7 +65,6 @@ def find_category(query: str):
         + "\n".join(f"• {e['category']}" for e in DATA)
     )
 
-
 def wants_commands(query: str) -> bool:
     keywords = [
         "command", "commands", "أوامر", "امر", "أمر",
@@ -82,7 +73,6 @@ def wants_commands(query: str) -> bool:
     ]
     return any(k in query.lower() for k in keywords)
 
-
 def format_description_only(entry) -> str:
     return (
         f"## 🛡️ {entry['category']}\n\n"
@@ -91,7 +81,6 @@ def format_description_only(entry) -> str:
         f"💡 اكتب **`{entry['category']} commands`** لعرض الأوامر"
     )
 
-
 def format_with_commands(entry) -> str:
     lines = [
         f"## 🛡️ {entry['category']}\n\n{entry['description']}\n\n---\n\n## ⚙️ الأوامر\n"
@@ -101,26 +90,23 @@ def format_with_commands(entry) -> str:
         lines.append(f"```bash\n{cmd['command']}\n```\n")
     return "\n".join(lines)
 
-
-# ✅ tuples format - works with all Gradio versions
-def respond(query: str, history: list):
-    if not query.strip():
+def respond(message: str, history: list):
+    if not message.strip():
         return history, ""
 
-    entry, err = find_category(query)
+    entry, err = find_category(message)
     answer = err if err else (
-        format_with_commands(entry) if wants_commands(query)
+        format_with_commands(entry) if wants_commands(message)
         else format_description_only(entry)
     )
 
+    # بناء التاريخ بالصيغة الجديدة
     history = history or []
-    history.append((query, answer))
+    history.append({"role": "user", "content": message})
+    history.append({"role": "assistant", "content": answer})
     return history, ""
 
-
-# ─────────────────────────────────────────
-# UI
-# ─────────────────────────────────────────
+# واجهة Gradio
 CATEGORIES_MD = "\n".join(f"- {e['category']}" for e in DATA)
 
 with gr.Blocks(title="CyberSec Assistant") as demo:
@@ -133,7 +119,7 @@ with gr.Blocks(title="CyberSec Assistant") as demo:
 
     with gr.Row():
         with gr.Column(scale=3):
-            chatbot = gr.Chatbot(label="المحادثة", height=500, type="tuples")
+            chatbot = gr.Chatbot(label="المحادثة", height=500)
             with gr.Row():
                 txt = gr.Textbox(
                     placeholder="اكتب اسم الثغرة مثل: XSS أو LFI أو SSRF ...",
@@ -154,11 +140,10 @@ with gr.Blocks(title="CyberSec Assistant") as demo:
                 """
             )
 
-    state = gr.State([])
-
-    send_btn.click(respond, [txt, state], [chatbot, txt])
-    txt.submit(respond, [txt, state], [chatbot, txt])
+    # ربط الأحداث
+    send_btn.click(respond, [txt, chatbot], [chatbot, txt])
+    txt.submit(respond, [txt, chatbot], [chatbot, txt])
 
     gr.Markdown("---\n*تصنيف: هجومي | Web Vulnerabilities | v1.2*")
 
-demo.launch()
+demo.launch()