Update app.py

app.py

@@ -1,18 +1,80 @@
-import os
+import gradio as gr
 from huggingface_hub import InferenceClient
 
-client = InferenceClient(
-    api_key=os.environ["HF_TOKEN"],
-)
-
-completion = client.chat.completions.create(
-    model="moonshotai/Kimi-K2-Instruct-0905:groq",
-    messages=[
-        {
-            "role": "user",
-            "content": "What is the capital of France?"
-        }
-    ],
-)
-
-print(completion.choices[0].message)
+def respond(
+    message, # هنا سيصبح message عبارة عن قاموس يحتوي على 'text' و 'files'
+    history: list[dict[str, str]],
+    system_message,
+    max_tokens,
+    temperature,
+    top_p,
+    hf_token: gr.OAuthToken,
+):
+    # التأكد من تسجيل الدخول
+    if hf_token is None:
+        yield "⚠️ يرجى تسجيل الدخول عبر Hugging Face أولاً."
+        return
+
+    client = InferenceClient(
+        model="moonshotai/Kimi-K2-Instruct", 
+        token=hf_token.token
+    )
+
+    messages = [{"role": "system", "content": system_message}]
+    messages.extend(history)
+
+    # معالجة الرسالة المدخلة (نص + ملفات)
+    user_text = message.get("text", "")
+    files = message.get("files", [])
+    
+    full_content = user_text
+    if files:
+        file_names = ", ".join([f.split('/')[-1] for f in files])
+        full_content += f"\n\n[المستخدم قام بإرفاق الملفات التالية: {file_names}]"
+        # ملاحظة: النماذج النصية لا "تقرأ" الملفات تلقائياً، 
+        # ستحتاج لمكتبات مثل PyMuPDF للـ PDF لاستخراج النص فعلياً.
+
+    messages.append({"role": "user", "content": full_content})
+
+    response = ""
+    try:
+        for msg in client.chat_completion(
+            messages,
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+        ):
+            if msg.choices and len(msg.choices) > 0:
+                token = msg.choices[0].delta.content
+                if token:
+                    response += token
+                    yield response
+    except Exception as e:
+        yield f"⚠️ حدث خطأ: {str(e)}"
+
+# إعداد واجهة Gradio مع دعم الوسائط المتعددة
+chatbot_ui = gr.Chatbot(type="messages")
+
+with gr.Blocks() as demo:
+    with gr.Sidebar():
+        gr.Markdown("### الإعدادات")
+        gr.LoginButton()
+        system_msg = gr.Textbox(value="أنت مساعد ذكي ومفيد.", label="System message")
+        max_t = gr.Slider(minimum=1, maximum=4096, value=1024, step=1, label="Max tokens")
+        temp = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.1, label="Temperature")
+        top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p")
+
+    gr.ChatInterface(
+        respond,
+        chatbot=chatbot_ui,
+        type="messages",
+        multimodal=True, # تفعيل دعم الملفات
+        additional_inputs=[system_msg, max_t, temp, top_p],
+        title="Kimi-K2 Multimodal Chatbot",
+        # تحديد صيغ الملفات المسموح بها
+        file_types=[".pdf", ".docx", ".zip"] 
+    )
+
+if __name__ == "__main__":
+    demo.launch()