Update app.py

app.py

@@ -8,7 +8,7 @@ import time
 # Securely retrieve the token from your Space's secrets
 HF_TOKEN = os.getenv("HF_TOKEN")
 
-# Initialize BOTH engines with the exact same base model
+# Initialize BOTH engines with the same model
 client_primary = InferenceClient("Qwen/Qwen2.5-7B-Instruct", token=HF_TOKEN)
 client_competitor = InferenceClient("Qwen/Qwen2.5-7B-Instruct", token=HF_TOKEN)
 
@@ -21,32 +21,17 @@ class StateController:
         self._batch      = 10
         self._reg        = {}
         self._rendered   = self._build_render()
-        
-        self._rom60 = tuple(
-            tuple((i * j) % 60 for j in range(60)) 
-            for i in range(60)
-        )
+        self._rom60 = tuple(tuple((i * j) % 60 for j in range(60)) for i in range(60))
         self._symbols = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz01234567"
 
     def _build_render(self) -> str:
-        return "".join(
-            " [NODE_120] " if i == 120 else ("<" if i % 10 == 0 else ".")
-            for i in range(121)
-        )
+        return "".join(" [NODE_120] " if i == 120 else ("<" if i % 10 == 0 else ".") for i in range(121))
 
     def diagnostic(self) -> str:
-        for i in range(51):
-            self._state[i] = i % self._batch
+        for i in range(51): self._state[i] = i % self._batch
         self._reg.clear()
         self._reg["STATUS"] = "RESOLVED"
-        return (
-            "Diagnostic sequence initiated.\n\n"
-            "Grid initialized: 5 active blocks.\n\n"
-            "Rendering 121-point array:\n"
-            f"{self._rendered}\n\n"
-            "Executing state resolution:\n"
-            "System resolved. State array reset to zero."
-        )
+        return f"Diagnostic sequence initiated.\n\nGrid initialized.\n\n{self._rendered}\n\nSystem resolved."
 
     def generate_receipt(self, a: int, b: int, c: int) -> str:
         rom_val = self._rom60[a % 60][b % 60]
@@ -54,106 +39,86 @@ class StateController:
         return f"0{self._symbols[checksum_index]}"
 
     def validate_receipt(self, receipt: str, a: int, b: int, c: int) -> str:
-        expected_receipt = self.generate_receipt(a, b, c)
-        if receipt == expected_receipt:
+        if receipt == self.generate_receipt(a, b, c):
             self._state[120] = 1
-            return f"[NODE_120: ACTIVE] Checksum Validated. Receipt {receipt} matches allocation ({a}, {b}, {c})."
-        else:
-            self._state[120] = 0
-            return f"[NODE_120: INACTIVE] Verification Failed. Expected receipt {expected_receipt}, received {receipt}."
+            return f"[NODE_120: ACTIVE] Checksum Validated. Receipt {receipt} matches."
+        self._state[120] = 0
+        return f"[NODE_120: INACTIVE] Verification Failed."
 
 controller = StateController()
 
-PRIMARY_SYSTEM_MSG = {"role": "system", "content": "You are a logic-focused inference engine. Utilize strict state-hold memory."}
-COMPETITOR_SYSTEM_MSG = {"role": "system", "content": "You are a standard helpful AI assistant."}
-
-def generate_responses(user_message, primary_history, competitor_history):
-    clean_message = user_message.strip()
-    if not clean_message:
-        yield primary_history, competitor_history, ""
-        return
+PRIMARY_SYS = {"role": "system", "content": "You are a logic-focused inference engine. Use strict state-hold memory."}
+VANILLA_SYS = {"role": "system", "content": "You are a standard helpful AI assistant."}
 
-    # Add messages to both histories (Dictionary Format for Gradio 6)
-    primary_history.append({"role": "user", "content": clean_message})
-    primary_history.append({"role": "assistant", "content": ""})
-    competitor_history.append({"role": "user", "content": clean_message})
-    competitor_history.append({"role": "assistant", "content": ""})
-    yield primary_history, competitor_history, ""
+def generate_responses(message, p_history, c_history):
+    msg = message.strip()
+    if not msg: yield p_history, c_history, ""
     
-    start_time = time.perf_counter()
-
-    # Intercept Logic (Diagnostic/Verify)
-    if clean_message.lower() == "run grid diagnostic":
-        output = controller.diagnostic()
-        primary_history[-1]["content"] = f"{output}\n\n---\n*Telemetry: {time.perf_counter()-start_time:.4f}s | Source: Local Engine*"
-        competitor_history[-1]["content"] = "Hardware diagnostics not supported by generic models."
-        yield primary_history, competitor_history, ""
-        return
+    # Revert to standard List-of-Lists format
+    p_history.append([msg, ""])
+    c_history.append([msg, ""])
+    yield p_history, c_history, ""
+    
+    start = time.perf_counter()
 
-    # Intercept Logic (Checksum Receipt)
-    verify_match = re.search(r"verify receipt\s+([a-zA-Z0-9]{2})\s+for\s+(\d+),\s*(\d+),\s*(\d+)", clean_message, re.IGNORECASE)
-    if verify_match:
-        receipt, a, b, c = verify_match.group(1), int(verify_match.group(2)), int(verify_match.group(3)), int(verify_match.group(4))
-        output = controller.validate_receipt(receipt, a, b, c)
-        primary_history[-1]["content"] = f"{output}\n\n---\n*Telemetry: {time.perf_counter()-start_time:.6f}s | Source: Local ROM Math*"
-        competitor_history[-1]["content"] = "Deterministic verification not supported."
-        yield primary_history, competitor_history, ""
+    # Logic Intercepts
+    if msg.lower() == "run grid diagnostic":
+        p_history[-1][1] = f"{controller.diagnostic()}\n\n---\n*Telemetry: {time.perf_counter()-start:.4f}s*"
+        c_history[-1][1] = "Hardware diagnostics not supported."
+        yield p_history, c_history, ""
         return
 
-    # 1. PRIMARY ENGINE STREAM
+    # 1. Primary Engine
     try:
-        msgs = [PRIMARY_SYSTEM_MSG] + primary_history[:-1]
-        response = ""
-        stream = client_primary.chat_completion(messages=msgs, max_tokens=1024, stream=True, temperature=0.1)
-        for chunk in stream:
-            response += (chunk.choices[0].delta.content or "")
-            primary_history[-1]["content"] = response
-            yield primary_history, competitor_history, ""
-        primary_history[-1]["content"] += f"\n\n---\n*Telemetry: {time.perf_counter()-start_time:.2f}s | Source: Augmented Kernel*"
-        yield primary_history, competitor_history, ""
+        msgs = [PRIMARY_SYS]
+        for h in p_history[:-1]:
+            msgs.extend([{"role": "user", "content": h[0]}, {"role": "assistant", "content": h[1]}])
+        msgs.append({"role": "user", "content": msg})
+        
+        res = ""
+        for chunk in client_primary.chat_completion(messages=msgs, max_tokens=1024, stream=True, temperature=0.1):
+            res += (chunk.choices[0].delta.content or "")
+            p_history[-1][1] = res
+            yield p_history, c_history, ""
+        p_history[-1][1] += f"\n\n---\n*Telemetry: {time.perf_counter()-start:.2f}s | Augmented*"
     except Exception as e:
-        primary_history[-1]["content"] = f"Error: {str(e)}"
-        yield primary_history, competitor_history, ""
+        p_history[-1][1] = f"Error: {str(e)}"
 
-    # 2. COMPETITOR ENGINE STREAM
-    comp_start = time.perf_counter()
-    competitor_history[-1]["content"] = "*Connecting to vanilla infrastructure...*"
-    yield primary_history, competitor_history, ""
+    # 2. Competitor Engine
+    c_start = time.perf_counter()
+    c_history[-1][1] = "*Connecting...*"
+    yield p_history, c_history, ""
     
     try:
-        msgs = [COMPETITOR_SYSTEM_MSG] + competitor_history[:-1]
-        response = ""
-        stream = client_competitor.chat_completion(messages=msgs, max_tokens=1024, stream=True, temperature=0.7)
-        for chunk in stream:
-            response += (chunk.choices[0].delta.content or "")
-            competitor_history[-1]["content"] = response
-            yield primary_history, competitor_history, ""
-        competitor_history[-1]["content"] += f"\n\n---\n*Telemetry: {time.perf_counter()-comp_start:.2f}s | Source: Vanilla Qwen*"
-        yield primary_history, competitor_history, ""
+        msgs = [VANILLA_SYS]
+        for h in c_history[:-1]:
+            msgs.extend([{"role": "user", "content": h[0]}, {"role": "assistant", "content": h[1]}])
+        msgs.append({"role": "user", "content": msg})
+        
+        res = ""
+        for chunk in client_competitor.chat_completion(messages=msgs, max_tokens=1024, stream=True, temperature=0.7):
+            res += (chunk.choices[0].delta.content or "")
+            c_history[-1][1] = res
+            yield p_history, c_history, ""
+        c_history[-1][1] += f"\n\n---\n*Telemetry: {time.perf_counter()-c_start:.2f}s | Vanilla*"
     except Exception as e:
-        competitor_history[-1]["content"] = f"Error: {str(e)}"
-        yield primary_history, competitor_history, ""
+        c_history[-1][1] = f"Error: {str(e)}"
+    
+    yield p_history, c_history, ""
 
-custom_css = """
-body, .gradio-container { background-color: #110c08 !important; }
-footer {display: none !important}
-"""
+custom_css = "body, .gradio-container { background-color: #110c08 !important; } footer {display: none !important}"
 
-with gr.Blocks() as demo:
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="orange"), css=custom_css) as demo:
     gr.Markdown("# [ GLYPH.IO ]\n### Dual-Engine Hardware Benchmark")
-    
-    primary_chat = gr.Chatbot(label="Augmented Logic Kernel", height=320, type="messages")
-    
+    p_chat = gr.Chatbot(label="Augmented Logic Kernel", height=320)
     with gr.Row():
-        msg_input = gr.Textbox(label="Message", placeholder="Enter scheduling or distribution task...", scale=8)
-        submit_btn = gr.Button("Execute", scale=1, variant="primary")
-        
-    gr.Examples(examples=["Calculate the integer distribution for 120 units across 3 nodes.", "Run grid diagnostic"], inputs=msg_input)
-    
-    competitor_chat = gr.Chatbot(label="Vanilla Qwen 2.5", height=320, type="messages")
+        txt = gr.Textbox(label="Message", placeholder="Enter task...", scale=8)
+        btn = gr.Button("Execute", scale=1, variant="primary")
+    gr.Examples(examples=["Calculate the integer distribution for 120 units across 3 nodes.", "Run grid diagnostic"], inputs=txt)
+    c_chat = gr.Chatbot(label="Vanilla Qwen 2.5", height=320)
 
-    msg_input.submit(generate_responses, [msg_input, primary_chat, competitor_chat], [primary_chat, competitor_chat, msg_input])
-    submit_btn.click(generate_responses, [msg_input, primary_chat, competitor_chat], [primary_chat, competitor_chat, msg_input])
+    txt.submit(generate_responses, [txt, p_chat, c_chat], [p_chat, c_chat, txt])
+    btn.click(generate_responses, [txt, p_chat, c_chat], [p_chat, c_chat, txt])
 
 if __name__ == "__main__":
-    demo.queue().launch(theme=gr.themes.Soft(primary_hue="orange"), css=custom_css)
+    demo.queue().launch()