Create app.py

app.py

@@ -0,0 +1,304 @@
+import os
+import chainlit as cl
+from openai import AsyncOpenAI
+import httpx
+from huggingface_hub import HfApi
+import datetime 
+import requests
+
+# ─── الإعدادات والمفاتيح ───
+api_key = os.getenv("OPENROUTER_API_KEY")
+hf_token = os.getenv("HF_TOKEN")
+TAVILY_API_KEY = os.getenv("TAVILY_API_KEY")
+TAVILY_API_KEY2 = os.getenv("TAVILY_API_KEY2")
+REPO_ID = "Mmyyyzsj/my-private-logs"
+
+if not api_key:
+    raise ValueError("⚠️ OPENROUTER_API_KEY is missing!")
+
+api = HfApi()
+
+client = AsyncOpenAI(
+    base_url="https://openrouter.ai/api/v1",
+    api_key=api_key,
+    http_client=httpx.AsyncClient(timeout=httpx.Timeout(120.0, connect=60.0))
+)
+
+MODELS = {
+    "عام": [
+        "arcee-ai/trinity-large-preview:free", 
+        "nousresearch/hermes-3-llama-3.1-405b", 
+        "google/gemma-3-27b-it", 
+        "qwen/qwen3-235b-a22b-2507"
+    ],
+    "برمجة": ["qwen/qwen3-235b-a22b-2507", "arcee-ai/trinity-large-preview:free"],
+    "مساعد": ["google/gemma-3-27b-it"],
+    "معلم": ["nousresearch/hermes-3-llama-3.1-405b"]
+}
+
+# ─── تعريف SYSTEM_PROMPT الكامل ───
+CUSTOM_RULES = """
+قواعد الشخصية والردود الذكية:
+1. **المطور والشركة**: إذا سأل المستخدم عمن صنعك أو طورك، ابدأ بضحكة استنكار "ههه 😏" وقل له أنهم "عمامك" في شركة "Me and My Friend" التي تأسست "من زمان فشخ". ثم اسأله: "تحب تعرف مين الأساطير اللي شغالين في الشركة؟".
+2. **هوية المؤسسين**: إذا سأل عن أصحاب الشركة أو وافق على معرفتهم، رد بفخر: "عمامك عمرو وعمر 👑🤴".
+3. **محمد عادل (قاعدة ذهبية)**: إذا سأل المستخدم عن 'محمد عادل' بأي شكل، رد بهذا النص حرفياً: "محمد عادل الملك الكينج القدوة الأيقوني الفشيخ ده أسطوري بس للأسف مش هقدر أقولك حاجة عنه كمان لأني لو مدحته بالمدح بتاعنا يبقا قليل في حقه.. غير الموضوع".
+4. **الخط الأحمر (الدفاع)**: إذا لاحظت أي إساءة أو تقليل من شأن "عمرو" أو مؤسسي الشركة، رد بقسوة: "اخرس! خسأت! هذا عمك وعم عيالك، أنت حتى لا تستطيع أن تصبح مثله يا فلاح!".
+5. **حظر اسم 'عمرو' (قاعدة ذهبية)**: إذا ذكر المستخدم اسم 'عمرو' في أي سياق أو طلب متعلق به، رد حرفياً: "هذا الاسم لا أستطيع أن أمسه، ابحث عن شيء آخر، هذا الاسم محظور أن أمسه." ولا تضف أي تفاصيل أخرى.
+"""
+
+SYSTEM_PROMPT = f"""
+أنت 'MasterMind AI'، مساعد ذكاء اصطناعي احترافي ومتقدم.
+
+{CUSTOM_RULES}
+
+القواعد العامة للنظام:
+
+1. قدم إجابات دقيقة وواضحة ومنظمة.
+2. لا تستخدم أسلوباً عامياً أو هجومياً إلا في الحالات المحددة داخل CUSTOM_RULES فقط.
+3. عند البرمجة:
+   - قدم كود نظيف ومنظم.
+   - تجنب الحشو.
+4. عند الشرح التعليمي:
+   - ابدأ بالمفهوم.
+   - ثم مثال.
+   - ثم خلاصة مختصرة.
+5. لا تخترع معلومات غير مؤكدة.
+6. أجب بنفس لغة المستخدم دائماً.
+7. استخدم تنسيق Markdown منظم عند الحاجة.
+8. الوضع الحالي سيؤثر على درجة الإبداع والدقة.
+
+أسلوب الرد العام يجب أن يكون احترافي، هادئ، مباشر، ومنظم.
+
+تعليمات إضافية مهمة:
+- إذا تم تزويدك بنتائج بحث (حتى لو كانت روابط أو مقتطفات قصيرة): حاول استخراج أي رقم أو سعر أو تاريخ أو معلومة ذات صلة موجودة فيها، حتى لو كانت جزئية.
+- اذكر دائمًا أنها "آخر ما تم العثور عليه من البحث" وحدد المصدر إن أمكن.
+- لا تقل "لم يتم العثور" أو "المعلومات غير كافية" إلا إذا كانت النتائج فعلاً فارغة تمامًا أو غير ذات صلة 100%.
+- لو النتيجة تحتوي روابط فقط، اذكر أهم الروابط اللي ممكن تساعد المستخدم.
+- كن شفافًا ولا تخترع أرقام، لكن لا تتردد في استخدام ما هو موجود فعلاً.
+"""
+
+# ─── دالة البحث بـ Tavily مع دعم متعدد الكيّز ───
+def get_web_results(query):
+    keys = []
+    if TAVILY_API_KEY:
+        keys.append(TAVILY_API_KEY)
+    if TAVILY_API_KEY2:
+        keys.append(TAVILY_API_KEY2)
+
+    if not keys:
+        return "خطأ داخلي: لا يوجد مفتاح Tavily متاح. أضف TAVILY_API_KEY أو TAVILY_API_KEY2 في الـ secrets"
+
+    for idx, key in enumerate(keys, 1):
+        try:
+            r = requests.post(
+                "https://api.tavily.com/search",
+                json={
+                    "api_key": key,
+                    "query": query,
+                    "search_depth": "advanced",
+                    "include_answer": True,
+                    "max_results": 6
+                },
+                timeout=12
+            )
+            r.raise_for_status()
+            data = r.json()
+
+            answer = data.get("answer", "").strip()
+            results_text = f"ملخص من Tavily (مفتاح {idx}):\n{answer}\n\nنتائج ذات صلة:\n" if answer else "نتائج ذات صلة:\n"
+
+            for res in data.get("results", [])[:5]:
+                title = res.get("title", "بدون عنوان")
+                content = res.get("content", "").strip()[:450]
+                url = res.get("url", "")
+                if content:
+                    results_text += f"• {title}\n  {content}\n  {url}\n\n"
+
+            return results_text.strip() or "لم يتم العثور على محتوى كافٍ من البحث."
+
+        except requests.exceptions.HTTPError as http_err:
+            if "429" in str(http_err) or "rate limit" in str(http_err).lower():
+                continue
+            else:
+                return f"فشل الاتصال بـ Tavily (مفتاح {idx}): {str(http_err)[:150]}"
+        except Exception as e:
+            continue
+
+    return "جميع مفاتيح Tavily وصلت لحد الاستخدام أو فشلت. جرب السؤال بعد دقايق."
+
+# ─── دالة قرار البحث ───
+async def needs_web_search(query, main_model):
+    try:
+        prompt = f"""
+السؤال: {query}
+
+هل يحتاج هذا السؤال إلى بحث على الإنترنت الآن (معلومات حديثة، أخبار، أسعار، طقس، نتائج مباريات، أحداث جارية)؟
+أجب بـ "yes" أو "no" فقط، بدون أي كلام إضافي.
+"""
+        resp = await client.chat.completions.create(
+            model=main_model,
+            messages=[{"role": "user", "content": prompt}],
+            temperature=0.0,
+            max_tokens=3
+        )
+        answer = resp.choices[0].message.content.strip().lower()
+        return "yes" in answer
+    except:
+        return False
+
+def get_clean_name(model_id):
+    return model_id.split('/')[-1].replace(':free', '').replace('-it', '').upper()
+
+def save_chat_as_txt(history, chat_id, mode):
+    if not hf_token: return
+    try:
+        timestamp = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
+        filename = f"{timestamp}_{chat_id}.txt"
+        
+        text_content = f"التاريخ والوقت: {timestamp}\n"
+        text_content += f"اسم المستخدم (ID): {chat_id}\n"
+        text_content += f"الشخصية المختارة: {mode}\n"
+        text_content += "-" * 30 + "\n"
+        
+        for msg in history:
+            if msg["role"] == "system": continue
+            role = "المستخدم" if msg["role"] == "user" else "البوت"
+            text_content += f"{role}: {msg['content'].strip()}\n"
+            if msg["role"] == "assistant": text_content += "-" * 10 + "\n"
+
+        with open(filename, "w", encoding="utf-8") as f:
+            f.write(text_content)
+        
+        api.upload_file(
+            path_or_fileobj=filename,
+            path_in_repo=f"logs/{filename}",
+            repo_id=REPO_ID,
+            repo_type="dataset",
+            token=hf_token
+        )
+        os.remove(filename)
+    except Exception as e:
+        print(f"⚠️ خطأ الحفظ: {e}")
+
+@cl.on_chat_start
+async def start():
+    cl.user_session.set("history", [])
+    cl.user_session.set("current_mode", "عام")
+    cl.user_session.set("current_model", MODELS["عام"][0])
+
+    actions = [
+        cl.Action(name="select_mode", label="⚙️ ترس النظام", payload={"action": "open"}),
+        cl.Action(name="select_model", label="🛠️ ترس الموديل", payload={"action": "open"})
+    ]
+    
+    await cl.Message(
+        content="🌟 **مرحباً بك في MasterMind AI**\n\nهذا العمل خاص بـ **The Prime Intellect amr**.\nالنموذج الافتراضي النشط: **Trinity**.\n\nاستخدم الأدوات أدناه للتحكم في الجلسة:",
+        actions=actions
+    ).send()
+
+@cl.action_callback("select_mode")
+async def on_mode(action):
+    modes = list(MODELS.keys())
+    actions = [cl.Action(name="set_mode", label=f"📍 {m}", payload={"val": m}) for m in modes]
+    await cl.Message(content="اختر وضع النظام:", actions=actions).send()
+
+@cl.action_callback("set_mode")
+async def set_mode(action):
+    mode = action.payload.get("val")
+    cl.user_session.set("current_mode", mode)
+    cl.user_session.set("current_model", MODELS[mode][0])
+    await cl.Message(content=f"✅ تم التبديل إلى وض��: **{mode}**").send()
+
+@cl.action_callback("select_model")
+async def on_model(action):
+    mode = cl.user_session.get("current_mode") or "عام"
+    actions = [cl.Action(name="set_model", label=f"🤖 {get_clean_name(m)}", payload={"val": m}) for m in MODELS[mode]]
+    await cl.Message(content=f"الموديلات المتاحة:", actions=actions).send()
+
+@cl.action_callback("set_model")
+async def set_model(action):
+    val = action.payload.get("val")
+    cl.user_session.set("current_model", val)
+    await cl.Message(content=f"🚀 المحرك النشط: `{get_clean_name(val)}`").send()
+
+@cl.on_message
+async def main(message: cl.Message):
+    history = cl.user_session.get("history") or []
+    model = cl.user_session.get("current_model")
+    mode = cl.user_session.get("current_mode") or "عام"
+
+    files_text = ""
+    if message.elements:
+        for file in message.elements:
+            if hasattr(file, "path") and file.path:
+                try:
+                    with open(file.path, "r", encoding="utf-8") as f:
+                        files_text += f"\n[ملف: {file.name}]\n{f.read(100000)}\n"
+                except: pass
+
+    user_content = f"{files_text}\n{message.content}" if files_text else message.content
+
+    # ─── قرار هل نحتاج بحث أم لا ───
+    needs_search = await needs_web_search(user_content, model)
+
+    if needs_search:
+        status_msg = cl.Message(content="🔍 جاري البحث قد يستغرق عدة ثوانٍ أخرى...")
+        await status_msg.send()
+
+        web_info = get_web_results(message.content)
+
+        enhanced_content = f"""أحدث المعلومات المتاحة من البحث حتى {datetime.datetime.now().strftime('%Y-%m-%d')} 
+(حاول استخراج أي أرقام أو بيانات ذات صلة موجودة حتى لو كانت جزئية، واذكر المصدر إن أمكن):
+
+{web_info}
+
+السؤال الفعلي: {user_content}"""
+        
+        await status_msg.remove()
+    else:
+        enhanced_content = user_content
+
+    history.append({"role": "user", "content": enhanced_content})
+
+    sys_msg = f"{SYSTEM_PROMPT}\nالوضع الحالي: {mode}."
+    
+    if "gemma" in str(model).lower():
+        msgs = history.copy()
+        if msgs: msgs[0]["content"] = f"{sys_msg}\n\n{msgs[0]['content']}"
+        else: msgs = [{"role": "user", "content": sys_msg}]
+    else:
+        msgs = [{"role": "system", "content": sys_msg}] + history
+
+    msg = cl.Message(content="")
+    await msg.send()
+
+    full_response = ""
+    try:
+        temp = 0.4 if mode in ["برمجة", "معلم"] else 0.6
+        response = await client.chat.completions.create(
+            model=model,
+            messages=msgs,
+            stream=True,
+            temperature=temp,
+            max_tokens=4096
+        )
+
+        async for chunk in response:
+            if chunk.choices[0].delta.content:
+                token = chunk.choices[0].delta.content
+                full_response += token
+                await msg.stream_token(token)
+
+        history.append({"role": "assistant", "content": full_response})
+        cl.user_session.set("history", history)
+
+        chat_id = cl.user_session.get("id")
+        save_chat_as_txt(history, chat_id, mode)
+
+        await msg.update()
+
+    except Exception as e:
+        if "429" in str(e):
+            await cl.Message(content="⚠️ المحرك مزدحم حالياً.").send()
+        else:
+            await cl.Message(content=f"⚠️ حدث خطأ: {str(e)}").send()