Update App.py

App.py

@@ -13,53 +13,49 @@ GEN_BY_ID = {m["id"]: m for m in GENERATORS}
 llm = pipeline("text-generation", model="gpt2", max_new_tokens=512)
 
 def call_llm(prompt):
-    out = llm(prompt, max_new_tokens=512, do_sample=False)[0]["generated_text"]
-    return out[len(prompt):].strip() if out.startswith(prompt) else out
+    o = llm(prompt, max_new_tokens=512, do_sample=False)[0]["generated_text"]
+    return o[len(prompt):].strip() if o.startswith(prompt) else o
 
-def generator_prompt(module_id, *inputs):
-    m = GEN_BY_ID[module_id]
+def generator_prompt(mid, *inputs):
+    m = GEN_BY_ID[mid]
     keys = list(m["input_placeholders"].keys())
-    inputs = {k: inputs[i] if i < len(inputs) else "" for i, k in enumerate(keys)}
-    sections = m["output_sections"]
+    vals = {k: inputs[i] if i < len(inputs) else "" for i, k in enumerate(keys)}
+    secs = m["output_sections"]
 
     p = []
-    p.append("You are a structured reasoning module.")
     p.append(f"MODULE: {m['label']}")
-    p.append("INPUTS:")
-    for k, v in inputs.items():
+    p.append("INPUT:")
+    for k, v in vals.items():
         p.append(f"{k.upper()}: {v}")
     p.append("")
-    p.append("Produce the following sections:")
-    for s in sections:
+    p.append("OUTPUT SECTIONS:")
+    for s in secs:
         p.append(f"- {s}")
     p.append("")
-    for s in sections:
+    for s in secs:
         p.append(f"{s}:")
         p.append("[content]")
         p.append("")
     return "\n".join(p)
 
-def checker_prompt(checker_id, *vals):
-    c = CHECKERS[checker_id]
+def checker_prompt(cid, *vals):
+    c = CHECKERS[cid]
     secs = c["output_sections"]
-    if len(vals) == 0:
-        orig, draft = "", ""
-    elif len(vals) == 1:
-        orig, draft = "", vals[0]
+    if len(vals) < 2:
+        orig, draft = "", vals[0] if vals else ""
     else:
-        *orig_parts, draft = vals
-        orig = "\n\n".join(orig_parts)
+        orig = "\n\n".join(vals[:-1])
+        draft = vals[-1]
 
     p = []
-    p.append("You are a strict evaluator.")
     p.append(f"CHECKER: {c['label']}")
     p.append("ORIGINAL TASK:")
     p.append(orig)
     p.append("")
-    p.append("DRAFT OUTPUT:")
+    p.append("DRAFT:")
     p.append(draft)
     p.append("")
-    p.append("Respond with sections:")
+    p.append("RESPOND WITH:")
     for s in secs:
         p.append(f"- {s}")
     p.append("")
@@ -77,32 +73,32 @@ def run_checker(cid, *inputs):
 
 def build_ui():
     with gr.Blocks(title="Modular Intelligence") as demo:
-        gr.Markdown("# Modular Intelligence\nSelect a module and generate output.")
+        gr.Markdown("# Modular Intelligence Demo")
 
         for m in GENERATORS:
             with gr.Tab(m["label"]):
                 gr.Markdown(m["description"])
-                input_boxes = []
-                for key, ph in m["input_placeholders"].items():
-                    tb = gr.Textbox(label=key, placeholder=ph, lines=4)
-                    input_boxes.append(tb)
+                inputs = []
+                for k, ph in m["input_placeholders"].items():
+                    t = gr.Textbox(label=k, placeholder=ph, lines=4)
+                    inputs.append(t)
 
-                output_box = gr.Textbox(label="Module output", lines=18)
-                gr.Button("Run module").click(
+                out = gr.Textbox(label="Output", lines=16)
+                gr.Button("Run").click(
                     fn=partial(run_generator, m["id"]),
-                    inputs=input_boxes,
-                    outputs=output_box
+                    inputs=inputs,
+                    outputs=out
                 )
 
                 if m.get("has_checker"):
                     cid = m.get("checker_id")
                     if cid in CHECKERS:
                         gr.Markdown("### Checker")
-                        checker_out = gr.Textbox(label="Checker output", lines=14)
-                        gr.Button("Run checker").click(
+                        chk = gr.Textbox(label="Checker Output", lines=14)
+                        gr.Button("Check").click(
                             fn=partial(run_checker, cid),
-                            inputs=input_boxes + [output_box],
-                            outputs=checker_out
+                            inputs=inputs + [out],
+                            outputs=chk
                         )
     return demo