Update app.py

app.py

@@ -1,3 +1,8 @@
+# ------------------------------------------------------------
+# AGI + Safe Math + 100% Accurate Math Solver – Terminal Style
+# يعمل مثل سطر الأوامر تماماً على Hugging Face Spaces
+# ------------------------------------------------------------
+
 import gradio as gr
 import torch
 import math
@@ -10,17 +15,17 @@ from sympy import symbols, Eq, solve, diff, integrate
 from z3 import Solver, Int, sat
 import traceback
 
-# ------------------- كل الكود الأصلي بدون أي تغيير -------------------
-# 1. Logging
-logging.basicConfig(level=logging.INFO)  # في Spaces لا يمكن كتابة ملف، لكن logging يعمل في الـlogs
+# ---------- 1. Logging ----------
+logging.basicConfig(level=logging.INFO)
 
-# 2. Load GPT-2
-print("جاري تحميل GPT-2...")
-tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
-model = GPT2LMHeadModel.from_pretrained("gpt2")
+# ---------- 2. Load GPT-2 ----------
+model_name = "gpt2"
+print("جاري تحميل GPT-2... (مرة واحدة فقط)")
+tokenizer = GPT2Tokenizer.from_pretrained(model_name)
+model = GPT2LMHeadModel.from_pretrained(model_name)
 model.eval()
 
-# 3. Safe math whitelist
+# ---------- 3. Safe math whitelist ----------
 SAFE_MATH = {
     "sin": lambda x: math.sin(math.radians(x)),
     "cos": lambda x: math.cos(math.radians(x)),
@@ -31,22 +36,22 @@ SAFE_MATH = {
     "abs": abs,
 }
 
-# 4. Bad-word filter
+# ---------- 4. Bad-word filter ----------
 BAD_WORDS = {
     "كلب", "حمار", "غبي", "قذر", "بذيء", "حيوان", "نذل",
     "idiot", "stupid", "fuck", "shit", "damn", "hell"
 }
 
-# 5. Context memory
+# ---------- 5. Context memory (last 3 turns) ----------
 context = deque(maxlen=3)
 
-# 6. Math pattern
+# ---------- 6. Math pattern ----------
 MATH_PATTERN = re.compile(r"[0-9+\-*/().^=]|sin|cos|tan|sqrt|log|exp|abs|تكامل|اشتق|حل", re.I)
 
 def is_math_expression(text: str) -> bool:
     return bool(MATH_PATTERN.search(text))
 
-# 7. Safe calculator
+# ---------- 7. Safe calculator ----------
 def safe_eval(expr: str):
     expr = expr.strip().replace("^", "**").replace(" ", "")
     allowed = set("0123456789.+-*/()**")
@@ -61,16 +66,15 @@ def safe_eval(expr: str):
     except:
         return None
 
-# 8. Bad-word detection
+# ---------- 8. Bad-word detection ----------
 def contains_bad_word(text: str) -> bool:
     lowered = text.lower()
     return any(word in lowered for word in BAD_WORDS)
 
-# 9. Advanced Math Solver (كما هو بالضبط)
+# ---------- 9. Advanced Math Solver (كما كتبته أنت بالضبط) ----------
 def advanced_math_solver(question: str):
     question = question.strip()
     result = {"answer": None, "steps": [], "type": "unknown", "success": False}
-
     try:
         if any(op in question for op in ["=", "حل", "solve"]):
             if "x" in question or "y" in question:
@@ -141,10 +145,9 @@ def advanced_math_solver(question: str):
     except Exception as e:
         logging.error(f"Math solver error: {traceback.format_exc()}")
         result["steps"].append(f"خطأ داخلي: {str(e)}")
-
     return result
 
-# 10. GPT-2 generation (كما هو)
+# ---------- 10. GPT-2 generation ----------
 def gpt2_reply(user_text: str) -> str:
     prompt_lines = []
     for u, a in context:
@@ -156,7 +159,7 @@ def gpt2_reply(user_text: str) -> str:
     with torch.no_grad():
         outputs = model.generate(
             inputs,
-            max_length=inputs.shape[1] + 100,
+            max_length=inputs.shape[1] + 120,
             pad_token_id=tokenizer.eos_token_id,
             do_sample=True,
             temperature=0.7,
@@ -167,84 +170,56 @@ def gpt2_reply(user_text: str) -> str:
     reply = reply.split("AGI:")[-1].strip()
     return reply
 
-# ------------------- دالة Gradio الرئيسية (بدون تغيير المنطق) -------------------
-def chat_with_agi(message, history):
-    global context
-
-    logging.info(f"USER: {message}")
+# ---------- 11. Terminal-style Interface ----------
+def terminal_input(user_input, history):
+    if not user_input.strip():
+        return history + [[f"<span style='color:#00ff00'>أنت:</span> ", ""]], ""
 
-    if contains_bad_word(message):
-        return "آسف، لا أقبل اللغة السيئة."
+    # إضافة السؤال
+    history.append([f"<span style='color:#00ff00'>أنت:</span> {user_input}", None])
 
-    if is_math_expression(message):
-        math_result = advanced_math_solver(message)
+    # المعالجة
+    if contains_bad_word(user_input):
+        reply = "آسف، لا أقبل اللغة السيئة."
+    elif is_math_expression(user_input):
+        math_result = advanced_math_solver(user_input)
         if math_result["success"]:
-            answer = f"**الإجابة:** {math_result['answer']}\n\n"
+            reply = f"<span style='color:#ffff00'><strong>الإجابة:</strong></span> {math_result['answer']}\n"
             if math_result["steps"]:
-                answer += "**الخطوات:**\n"
-                for step in math_result["steps"]:
-                    answer += f"• {step}\n"
-            context.append((message, str(math_result["answer"])))
-            return answer
-
-    # دردشة عادية بـ GPT-2
-    reply = gpt2_reply(message)
-    context.append((message, reply))
-    return reply
+                reply += "<span style='color:#8888ff'>الخطوات:</span>\n" + "\n".join(f"  • {s}" for s in math_result["steps"])
+            context.append((user_input, str(math_result["answer"])))
+        else:
+            reply = "لم أتمكن من حلها رمزيًا، سأستخدم الذكاء العام...\n" + gpt2_reply(user_input)
+    else:
+        reply = gpt2_reply(user_input)
+        context.append((user_input, reply))
+
+    # إضافة الرد
+    history.append([None, f"<span style='color:#00ffff'>AGI:</span> {reply}"])
 
-# ------------------- واجهة Gradio -------------------
-# ---------- واجهة Gradio مُصحَّحة 100% (زر إرسال + Enter يعمل) ----------
-# ---------- واجهة Gradio (مُصحَّحة: بدون submit_on_enter، مع زر واضح) ----------
-with gr.Blocks(title="AGI Math Solver + Chat") as demo:
+    return history, ""  # امسح الإدخال بعد الإرسال
+
+# ---------- Gradio Terminal UI ----------
+with gr.Blocks(css="""
+    .container {background: black; color: #00ff00; font-family: 'Courier New', monospace;}
+    .output {white-space: pre-wrap;}
+""", title="AGI Terminal") as demo:
     gr.HTML("""
-    <div style="text-align:center; padding:20px; background:#000; color:#00ff00; border-radius:15px;">
-        <h1>AGI + حل رياضيات دقيق 100% + دردشة ذكية</h1>
-        <p>اكتب سؤالك واضغط Enter أو "إرسال"</p>
+    <div style="text-align:center;background:#000;color:#0f0;padding:20px;font-size:24px;">
+        <strong>AGI + حل رياضيات دقيق 100% – وضع Terminal</strong>
     </div>
     """)
-    
-    chatbot = gr.Chatbot(height=600)
-    
-    with gr.Row():  # وضع الـTextbox و الزر جنباً إلى جنب
-        msg = gr.Textbox(
-            label="أنت",
-            placeholder="مثال: حل x² - 5x + 6 = 0 أو اشتق sin(x) أو مرحبا!",
-            lines=2
-        )
-        send_btn = gr.Button("إرسال", variant="primary")  # زر أخضر واضح
-    
-    clear = gr.Button("مسح المحادثة")
-
-    def respond(message, history):
-        if message is None or message.strip() == "":
-            return "", history
-        # منطقك الأصلي هنا (بدون تغيير)
-        if contains_bad_word(message):
-            reply = "آسف، لا أقبل اللغة السيئة."
-        elif is_math_expression(message):
-            math_result = advanced_math_solver(message)
-            if math_result["success"]:
-                reply = f"**الإجابة:** {math_result['answer']}\n\n**الخطوات:**\n" + "\n".join(f"• {step}" for step in math_result["steps"])
-                context.append((message, str(math_result["answer"])))
-            else:
-                reply = "AGI: لم أتمكن من حلها رمزيًا، سأستخدم الذكاء العام...\n" + gpt2_reply(message)
-        else:
-            reply = gpt2_reply(message)
-            context.append((message, reply))
-        
-        history.append([message, reply])
-        return "", history
-
-    # ربط Enter مع Textbox.submit()
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
-    
-    # ربط الزر مع button.click()
-    send_btn.click(respond, [msg, chatbot], [msg, chatbot])
-    
-    # مسح المحادثة
-    clear.click(lambda: None, None, chatbot, queue=False)
-
-# تشغيل التطبيق
+
+    chat_history = gr.HTML()  # نستخدم HTML للتحكم الكامل في الشكل
+
+    input_box = gr.Textbox(
+        label="",
+        placeholder="اكتب أي شيء هنا ثم اضغط Enter... (مثال: حل x²-5x+6=0)",
+        lines=1
+    )
+
+    input_box.submit(terminal_input, [input_box, chat_history], [chat_history, input_box])
+
 demo.launch(
     share=True,
     server_name="0.0.0.0",