Upload app.py with huggingface_hub

app.py

@@ -20,7 +20,7 @@ def projectile_calc(v0_mps: float, theta_deg: float, y0_m: float, g: float, weig
     if not (0 < v0_mps <= 500): errors.append("Initial speed must be in (0, 500] m/s.")
     if not (-10 <= theta_deg <= 90): errors.append("Launch angle must be between -10° and 90°.")
     if not (0 <= y0_m <= 1000): errors.append("Initial height must be in [0, 1000] m.")
-    if not (1 <= g <= 50): errors.append("Gravity must be in [1, 50] m/s².")
+    if not (1 <= g <= 50): errors.append("Gravity must be in [1, 50] m/s^2.")
     if not (0.05 <= weight_kg <= 50): errors.append("Ball weight must be in [0.05, 50] kg.")
     if errors:
         return {"ok": False, "errors": errors}
@@ -73,6 +73,7 @@ def llm_one_sentence(structured: Dict[str, Any]) -> str:
     i = structured["inputs"]
     d = structured["derived"]
     o = structured["outputs"]
+
     payload = {
         "inputs": {
             "v0_mps": round(i["v0_mps"], 3),
@@ -91,14 +92,54 @@ def llm_one_sentence(structured: Dict[str, Any]) -> str:
             "impact_speed_mps": round(o["impact_speed_mps"], 3),
         }
     }
-    instruction = dedent("""        Produce EXACTLY one sentence of this form using only the JSON values:
-        "If you threw a ball that weighed {weight_kg} kg at v0={v0_mps} m/s, θ={theta_deg}°, from y0={y0_m} m under g={g_mps2} m/s²,
-        it would stay in the air for {time_of_flight_s} s, reach {max_height_m} m, travel {range_m} m, and impact at {impact_speed_mps} m/s."
-    """)
-    user_prompt = f"JSON:\n{json.dumps(payload, indent=2)}\n\nFollow the instruction exactly."
-    prompt = f"<|system|>\n{SYSTEM_PROMPT}\n<|user|>\n{instruction}\n\n{user_prompt}\n<|assistant|>"
-    out = llm(prompt, max_new_tokens=120, do_sample=False, temperature=0.0, return_full_text=False)
-    return out[0]["generated_text"].strip()
+
+    system_msg = (
+        "You are a careful technical writer. "
+        "Write EXACTLY ONE sentence using ONLY numbers provided in the JSON. "
+        "Do not invent or rename fields; keep units as given; use ~2–3 sig figs."
+    )
+
+    # Show the exact sentence shape we want.
+    instruction = '''Use only these keys: inputs, derived, outputs.
+    Return exactly one sentence in this form (fill in the braces with the JSON values):
+
+    If you threw a ball that weighed {weight_kg} kg at v0={v0_mps} m/s, θ={theta_deg}, from y0={y0_m} m under g={g_mps2} m/s^2,
+    it would stay in the air for {time_of_flight_s} s, reach {max_height_m} m, travel {range_m} m, and impact at {impact_speed_mps} m/s.
+    '''
+
+    user_msg = f"JSON:
+{json.dumps(payload, indent=2)}
+Produce the single sentence as specified."
+
+    # ✅ Use the tokenizer's chat template
+    messages = [
+        {"role": "system", "content": system_msg},
+        {"role": "user", "content": instruction},
+        {"role": "user", "content": user_msg},
+    ]
+    prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
+
+    out = llm(
+        prompt,
+        max_new_tokens=96,
+        do_sample=False,
+        temperature=0.0,
+        return_full_text=False,
+    )
+    text = out[0]["generated_text"].strip()
+
+    # Minimal fallback: if the model didn't produce a sensible one-liner, synthesize it.
+    if ("If you threw a ball" not in text) or (len(text.split()) < 6):
+        p = payload
+        text = (
+            f"If you threw a ball that weighed {p['inputs']['weight_kg']} kg at v0={p['inputs']['v0_mps']} m/s, "
+            f"θ={p['inputs']['theta_deg']}, from y0={p['inputs']['y0_m']} m under g={p['inputs']['g_mps2']} m/s^2, "
+            f"it would stay in the air for {p['outputs']['time_of_flight_s']} s, "
+            f"reach {p['outputs']['max_height_m']} m, travel {p['outputs']['range_m']} m, "
+            f"and impact at {p['outputs']['impact_speed_mps']} m/s."
+        )
+
+    return text
 
 def run_once(v0_mps, theta_deg, y0_m, g_mps2, weight_kg):
     rec = projectile_calc(float(v0_mps), float(theta_deg), float(y0_m), float(g_mps2), float(weight_kg))
@@ -111,7 +152,7 @@ def run_once(v0_mps, theta_deg, y0_m, g_mps2, weight_kg):
         "v0_mps [m/s]": round(i["v0_mps"], 3),
         "theta_deg [deg]": round(i["theta_deg"], 3),
         "y0_m [m]": round(i["y0_m"], 3),
-        "g_mps2 [m/s²]": round(i["g_mps2"], 3),
+        "g_mps2 [m/s^2]": round(i["g_mps2"], 3),
         "weight_kg [kg]": round(i["weight_kg"], 3),
         "t_apex_s [s]": round(d["t_apex_s"], 4),
         "max_height_m [m]": round(o["max_height_m"], 3),
@@ -130,14 +171,14 @@ with gr.Blocks(title="Projectile Motion — Deterministic + One-Sentence LLM") a
         v0 = gr.Slider(1, 200, value=30.0, step=0.5, label="Initial speed v₀ [m/s]")
         theta = gr.Slider(-10, 90, value=45.0, step=0.5, label="Launch angle θ [deg]")
         y0 = gr.Slider(0, 20, value=1.5, step=0.1, label="Initial height y₀ [m]")
-        g = gr.Slider(5, 20, value=9.81, step=0.01, label="Gravity g [m/s²]")
+        g = gr.Slider(5, 20, value=9.81, step=0.01, label="Gravity g [m/s^2]")
         w = gr.Slider(0.05, 10.0, value=0.43, step=0.01, label="Ball weight [kg]")
 
     run_btn = gr.Button("Compute")
 
     results_df = gr.Dataframe(
         headers=[
-            "v0_mps [m/s]", "theta_deg [deg]", "y0_m [m]", "g_mps2 [m/s²]", "weight_kg [kg]",
+            "v0_mps [m/s]", "theta_deg [deg]", "y0_m [m]", "g_mps2 [m/s^2]", "weight_kg [kg]",
             "t_apex_s [s]", "max_height_m [m]", "time_of_flight_s [s]", "range_m [m]", "impact_speed_mps [m/s]"
         ],
         label="All values (inputs + derived)",