Upload app 2.py

app 2.py

@@ -0,0 +1,157 @@
+import gradio as gr
+import json, os, time, uuid
+from smolagents import CodeAgent, tool
+from smolagents.models import LiteLLMModel
+try:
+    from smolagents import HfApiModel
+except Exception:
+    HfApiModel = None
+
+# --------- Shared model factory ---------
+def get_model():
+    if os.getenv("OPENAI_API_KEY"):
+        return LiteLLMModel(model_id=os.getenv("OPENAI_MODEL","gpt-5"),
+                            api_key=os.getenv("OPENAI_API_KEY"))
+    elif HfApiModel is not None:
+        return HfApiModel(os.getenv("HF_MODEL","Qwen/Qwen2.5-7B-Instruct"))
+    else:
+        raise RuntimeError("No model available. Set OPENAI_API_KEY or install HF Inference support.")
+
+# =======================================
+# Exercise 1: Hello Tools (add + multiply)
+# =======================================
+@tool
+def add(a: float, b: float) -> float:
+    """Return a+b"""
+    return a + b
+
+@tool
+def mul(a: float, b: float) -> float:
+    """Return a*b"""
+    return a * b
+
+def run_ex1():
+    system = (
+        "You are a careful math assistant. Use the available tools to compute the answer.\n"
+        "Finally, PRINT ONLY a JSON object like {\"result\": <number>}."
+    )
+    agent = CodeAgent(
+        tools=[add, mul],
+        model=get_model(),
+        system_prompt=system,
+        add_base_tools=False,
+        verbosity_level=1,
+        temperature=0.0,
+    )
+    goal = "Compute 2 * (3 + 4) and return {\"result\": 14}."
+    out = agent.run(goal)
+    return str(out)
+
+# =======================================
+# Exercise 2: Guardrails + fixed plan
+# =======================================
+def run_ex2():
+    system = (
+        "Follow this fixed plan exactly:\n"
+        "1) Call add(a=3, b=4) and store the result in x.\n"
+        "2) Call mul(a=2, b=x) to compute y.\n"
+        "3) PRINT ONLY this JSON: {\"result\": y}\n"
+        "Never print explanations."
+    )
+    agent = CodeAgent(
+        tools=[add, mul],
+        model=get_model(),
+        system_prompt=system,
+        add_base_tools=False,
+        verbosity_level=1,
+        temperature=0.0,
+    )
+    goal = "Please give me the product of two and the sum of three and four — as JSON."
+    out = agent.run(goal)
+    return str(out)
+
+# =======================================
+# Exercise 3: Mini PR -> PO
+# =======================================
+@tool
+def validate_pr(pr: dict) -> dict:
+    """Return {ok: bool, errors: [...]} after checking pr_id, requester, cost_center, currency, items."""
+    req = ["pr_id","requester","cost_center","currency","items"]
+    errors = [f"Missing {k}" for k in req if k not in pr]
+    if not isinstance(pr.get("items", []), list) or not pr["items"]:
+        errors.append("Items must be a non-empty list")
+    return {"ok": len(errors)==0, "errors": errors}
+
+@tool
+def create_po(pr: dict) -> dict:
+    """Create a simple PO JSON with totals and a generated po_id."""
+    subtotal = 0.0
+    items_out = []
+    for it in pr.get("items", []):
+        line = float(it["quantity"]) * float(it["unit_price"])
+        items_out.append({"sku": it["sku"], "line_total": round(line,2)})
+        subtotal += line
+    tax = round(subtotal * 0.08, 2)
+    total = round(subtotal + tax, 2)
+    return {
+        "po_id": f"PO-{int(time.time())}-{uuid.uuid4().hex[:6].upper()}",
+        "currency": pr.get("currency","SGD"),
+        "items": items_out,
+        "subtotal": round(subtotal,2),
+        "tax": tax,
+        "total": total,
+        "source_pr_id": pr.get("pr_id")
+    }
+
+DEFAULT_PR = json.dumps({
+    "pr_id": "PR-1001",
+    "requester": "A. Tan",
+    "cost_center": "CC-SG-OPS",
+    "currency": "SGD",
+    "items": [
+        {"sku":"PAPER-A4-80G","quantity":20,"unit_price":6.8},
+        {"sku":"STAPLER-01","quantity":5,"unit_price":18.5}
+    ]
+}, indent=2)
+
+def run_ex3(pr_text):
+    try:
+        pr = json.loads(pr_text)
+    except Exception as e:
+        return f"Invalid JSON: {e}"
+    system = (
+        "You are a procurement agent. Use the tools as follows:\n"
+        "1) validate_pr(pr). If ok==false, PRINT ONLY {\"error\": errors}.\n"
+        "2) If ok==true, call create_po(pr).\n"
+        "3) PRINT ONLY the resulting PO JSON. No extra text."
+    )
+    agent = CodeAgent(
+        tools=[validate_pr, create_po],
+        model=get_model(),
+        system_prompt=system,
+        add_base_tools=False,
+        verbosity_level=1,
+        temperature=0.0,
+    )
+    goal = "Convert this PR to a PO (strict JSON). PR is in additional_context['pr']."
+    out = agent.run(goal, additional_context={"pr": pr})
+    return str(out)
+
+with gr.Blocks(title="Smolagents Beginner Lab (Online)") as demo:
+    gr.Markdown("# Smolagents Beginner Lab (Online)")
+    gr.Markdown("Click the buttons to run the exercises in the cloud. Set `OPENAI_API_KEY` in Space secrets for best results.")
+    with gr.Tab("1) Hello Tools"):
+        btn1 = gr.Button("Run Exercise 1")
+        out1 = gr.Textbox(label="Output", lines=6)
+        btn1.click(lambda: run_ex1(), inputs=None, outputs=out1)
+    with gr.Tab("2) Guardrails (deterministic)"):
+        btn2 = gr.Button("Run Exercise 2")
+        out2 = gr.Textbox(label="Output", lines=6)
+        btn2.click(lambda: run_ex2(), inputs=None, outputs=out2)
+    with gr.Tab("3) Mini PR → PO"):
+        pr_input = gr.Textbox(label="PR JSON", value=DEFAULT_PR, lines=16)
+        btn3 = gr.Button("Run Exercise 3")
+        out3 = gr.Textbox(label="Output", lines=10)
+        btn3.click(run_ex3, inputs=pr_input, outputs=out3)
+
+demo.launch()