LLM prompt formatted as a natural language

tray_sim.py

@@ -252,15 +252,14 @@ def run_tray_simulation(seed=0, num_objects=N_OBJECTS, azimuth=45, elevation=-25
     with open(json_path, "w") as f:
         json.dump(physics_state, f, indent=2)
 
-    llm_friendly_output = format_llm_prompt(physics_state)
-
-    #formatted = format_llm_prompt(physics_state)
-    #prompt = format_as_natural_language_prompt(
-    #    formatted,
-    #    task_description="Explain which objects are likely to fall if the tray is tilted slightly to the right."
-    #)
-    #print(prompt)
+    # LLM-friendly prompt output 
+    #prompt = format_llm_prompt(physics_state)
 
+    formatted = format_llm_prompt(physics_state)
+    prompt = format_as_natural_language_prompt(
+        formatted,
+        task_description="Explain which objects are likely to fall if the tray is tilted slightly to the right."
+    )
 
     # Save to GIF
     gif_path = os.path.join(tempfile.gettempdir(), f"tray_sim_{seed}.gif")
@@ -269,4 +268,4 @@ def run_tray_simulation(seed=0, num_objects=N_OBJECTS, azimuth=45, elevation=-25
     #return gif_path
     #return gif_path, stability_flags
     #return gif_path, stability_flags, physics_state  # optionally also return json_path
-    return gif_path, stability_flags, physics_state, llm_friendly_output, json_path # optionally also return json_path
+    return gif_path, stability_flags, physics_state, prompt, json_path # optionally also return json_path