Update app.py

app.py

@@ -1,22 +1,91 @@
-import os
+import gradio as gr
 from openai import OpenAI
+import zipfile
+import os
+import fitz  # PyMuPDF
+import docx
 
-# قراءة المفتاح من السيكرتس الخاصة بـ Hugging Face
-MOONSHOT_API_KEY = os.environ.get("MOONSHOT_API_KEY")
+# جلب المفتاح من إعدادات Secrets
+API_KEY = os.environ.get("MOONSHOT_API_KEY")
 
-def get_client():
-    if not MOONSHOT_API_KEY:
-        return None
-    return OpenAI(
-        api_key=MOONSHOT_API_KEY,
-        base_url="https://api.moonshot.cn/v1",
-    )
+def extract_text_from_file(file_path):
+    try:
+        filename = os.path.basename(file_path)
+        ext = filename.split('.')[-1].lower()
+        if ext == 'pdf':
+            with fitz.open(file_path) as doc:
+                return f"\n[PDF: {filename}]\n" + "".join([p.get_text() for p in doc])[:30000]
+        elif ext == 'docx':
+            doc_obj = docx.Document(file_path)
+            return f"\n[Word: {filename}]\n" + "\n".join([p.text for p in doc_obj.paragraphs])[:30000]
+        elif ext == 'zip':
+            report = f"\n[ZIP: {filename}]\n"
+            with zipfile.ZipFile(file_path, 'r') as z:
+                files = z.namelist()
+                report += f"يحتوي المجلد على {len(files)} ملف. فهرس أول 50 ملف:\n" + "\n".join(files[:50])
+            return report
+        else:
+            with open(file_path, "r", encoding="utf-8", errors="ignore") as f:
+                return f"\n[File: {filename}]\n" + f.read()[:20000]
+    except Exception as e:
+        return f"\n⚠️ خطأ في قراءة {os.path.basename(file_path)}: {str(e)}\n"
 
 def respond(message, history, system_message, max_tokens, temperature):
-    client = get_client()
-    
-    if client is None:
-        yield "⚠️ خطأ: لم يتم العثور على MOONSHOT_API_KEY في إعدادات Secrets."
+    if not API_KEY:
+        yield "⚠️ خطأ: يرجى إضافة MOONSHOT_API_KEY في Secrets."
         return
 
-    # ... باقي الكود الخاص بالإرسال والملفات
+    client = OpenAI(api_key=API_KEY, base_url="https://api.moonshot.cn/v1")
+    
+    user_text = message.get("text", "")
+    files = message.get("files", [])
+    
+    context = ""
+    if files:
+        yield "⏳ جاري تحليل ملفاتك بواسطة Kimi-K2..."
+        for f in files:
+            path = f if isinstance(f, str) else f.name
+            context += extract_text_from_file(path)
+
+    messages = [{"role": "system", "content": system_message}]
+    for msg in history:
+        messages.append(msg)
+    
+    full_user_msg = f"{context}\n\nالسؤال: {user_text}"
+    messages.append({"role": "user", "content": full_user_msg})
+
+    try:
+        completion = client.chat.completions.create(
+            model="moonshot-v1-128k",
+            messages=messages,
+            temperature=temperature,
+            max_tokens=max_tokens,
+            stream=True,
+        )
+        
+        response = ""
+        for chunk in completion:
+            content = chunk.choices[0].delta.content
+            if content:
+                response += content
+                yield response
+    except Exception as e:
+        yield f"⚠️ خطأ من سيرفر Moonshot: {str(e)}"
+
+# واجهة المستخدم
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🤖 Kimi-K2 (Moonshot AI) File Analyzer")
+    gr.ChatInterface(
+        respond,
+        multimodal=True,
+        type="messages",
+        additional_inputs=[
+            gr.Textbox(value="أنت مساعد ذكي يعتمد على نموذج Kimi-K2.", label="نظام المساعد"),
+            gr.Slider(512, 16384, value=4096, label="Max Tokens"),
+            gr.Slider(0.1, 1.5, value=0.7, label="Temperature"),
+        ]
+    )
+
+# التعديل الجوهري للتشغيل في Hugging Face Spaces
+if __name__ == "__main__":
+    demo.queue().launch(server_name="0.0.0.0", server_port=7860)