allow a feedback question, remove 2nd prompt

agent.py

@@ -3,7 +3,7 @@ import os
 from dotenv import load_dotenv
 import dspy
 
-import prompts as prompts_mod
+import prompts
 
 
 class BIDSifierAgent:
@@ -13,13 +13,13 @@ class BIDSifierAgent:
 		load_dotenv()
 		
 		if provider=="openai":
-			if model == "gpt-5":
+			if model == "gpt-5": #reasoning model that requires special handling
 				temperature = 1.0
 				lm = dspy.LM(f"{provider}/{model}", api_key=os.getenv("OPENAI_API_KEY"), temperature = temperature, max_tokens = 40000)
 			else:
-				lm = dspy.LM(f"{provider}/{model}", api_key=os.getenv("OPENAI_API_KEY"), temperature = temperature)
+				lm = dspy.LM(f"{provider}/{model}", api_key=os.getenv("OPENAI_API_KEY"), temperature = temperature, max_tokens = 10000)
 		else:
-			lm = dspy.LM(f"{provider}/{model}", api_key="")
+			lm = dspy.LM(f"{provider}/{model}", api_key="", max_tokens=10000)
 
       
 
@@ -35,34 +35,27 @@ class BIDSifierAgent:
 		output_root = context.get("output_root", "./bids_output")
 
 		if step == "summary":
-			return prompts_mod.summarize_dataset_prompt(
-				dataset_xml=dataset_xml,
-				readme_text=readme_text,
-				publication_text=publication_text,
-			)
-		if step == "create_root":
-			return prompts_mod.create_root_prompt(
-				output_root=output_root,
+			return prompts.summarize_dataset_prompt(
 				dataset_xml=dataset_xml,
 				readme_text=readme_text,
 				publication_text=publication_text,
 			)
 		if step == "create_metadata":
-			return prompts_mod.create_metadata_prompt(
+			return prompts.create_metadata_prompt(
 				output_root=output_root,
 				dataset_xml=dataset_xml,
 				readme_text=readme_text,
 				publication_text=publication_text,
 			)
 		if step == "create_structure":
-			return prompts_mod.create_structure_prompt(
+			return prompts.create_structure_prompt(
 				output_root=output_root,
 				dataset_xml=dataset_xml,
 				readme_text=readme_text,
 				publication_text=publication_text,
 			)
 		if step == "rename_move":
-			return prompts_mod.rename_and_move_prompt(
+			return prompts.rename_and_move_prompt(
 				output_root=output_root,
 				dataset_xml=dataset_xml,
 				readme_text=readme_text,
@@ -71,7 +64,7 @@ class BIDSifierAgent:
 		raise ValueError(f"Unknown step: {step}")
 
 	def run_step(self, step: str, context: Dict[str, Any]) -> str:
-		system_msg = prompts_mod.system_prompt()
+		system_msg = prompts.system_prompt()
 		user_msg = self._build_user_prompt(step, context)
 		resp = self.llm(
 			messages=[
@@ -82,6 +75,17 @@ class BIDSifierAgent:
 		)
 		return resp[0]
 
+	def run_query(self, query: str) -> str:
+		system_msg = prompts.system_prompt()
+		resp = self.llm(
+			messages=[
+				{"role": "system", "content": system_msg},
+				{"role": "user", "content": query},
+			],
+			temperature=self.temperature,
+		)
+		return resp[0]
+
 
 __all__ = ["BIDSifierAgent"]
 

cli.py

@@ -36,7 +36,11 @@ def _print_commands(commands: List[str]) -> None:
     if not commands:
         print("(No commands detected in fenced bash block.)")
         return
-    print("\nProposed commands (NOT executed):")
+    print("-----"*10)
+
+    print("COMMANDS TO EXECUTE:")
+    
+    print("-----"*10)
     for c in commands:
         print(f"  {c}")
 
@@ -53,7 +57,15 @@ def short_divider(title: str) -> None:
     print("\n" + "=" * 80)
     print(title)
     print("=" * 80 + "\n")
-
+    
+def enter_feedback_loop(agent: BIDSifierAgent, context: dict) -> dict:
+    feedback = input("\nAny comments or corrections to the summary? (press Enter to skip): ").strip()
+    while feedback:
+        context["user_feedback"] += feedback
+        agent_response = agent.run_query(feedback)
+        print(agent_response)
+        feedback = input("\nAny additional comments or corrections? (press Enter to skip): ").strip()
+    return context
 
 def main(argv: Optional[List[str]] = None) -> int:
     parser = argparse.ArgumentParser(
@@ -82,6 +94,7 @@ def main(argv: Optional[List[str]] = None) -> int:
         "readme_text": readme_text,
         "publication_text": publication_text,
         "output_root": args.output_root,
+        "user_feedback": "",
     }
 
     command_env = {
@@ -99,40 +112,36 @@ def main(argv: Optional[List[str]] = None) -> int:
     short_divider("Step 1: Understand dataset")
     summary = agent.run_step("summary", context)
     print(summary)
+    context = enter_feedback_loop(agent, context)
     if not prompt_yes_no("Proceed to create BIDS root?", default=True):
         return 0
-
-    short_divider("Step 2: Propose commands to create BIDS root")
-    root_plan = agent.run_step("create_root", context)
-    print(root_plan)
-    cmds = parse_commands_from_markdown(root_plan)
-    _print_commands(cmds)
-    if not prompt_yes_no("Proceed to create metadata files?", default=True):
-        return 0
-
-    short_divider("Step 3: Propose commands to create metadata files")
+    
+    short_divider("Step 2: Propose commands to create metadata files")
     meta_plan = agent.run_step("create_metadata", context)
     print(meta_plan)
     cmds = parse_commands_from_markdown(meta_plan)
     _print_commands(cmds)
+    context = enter_feedback_loop(agent, context)
     if not prompt_yes_no("Proceed to create empty BIDS structure?", default=True):
         return 0
 
-    short_divider("Step 4: Propose commands to create dataset structure")
+    short_divider("Step 3: Propose commands to create dataset structure")
     struct_plan = agent.run_step("create_structure", context)
     print(struct_plan)
     cmds = parse_commands_from_markdown(struct_plan)
     _print_commands(cmds)
+    context = enter_feedback_loop(agent, context)
     if not prompt_yes_no("Proceed to propose renaming/moving?", default=True):
         return 0
 
-    short_divider("Step 5: Propose commands to rename/move files")
+    short_divider("Step 4: Propose commands to rename/move files")
     move_plan = agent.run_step("rename_move", context)
     print(move_plan)
     cmds = parse_commands_from_markdown(move_plan)
     _print_commands(cmds)
+    context = enter_feedback_loop(agent, context)
 
-    print("\nAll steps completed. Commands were only displayed (never executed). Use them manually or in a future Gradio/HF Space interface.")
+    print("\nAll steps completed. Commands were only displayed - use them manually")
     return 0
 
 

prompts.py

@@ -33,7 +33,7 @@ def system_prompt() -> str:
 
 def summarize_dataset_prompt(*, dataset_xml: Optional[str], readme_text: Optional[str], publication_text: Optional[str]) -> str:
 	return f"""
-Step 1/5 — Understand the dataset and produce a short summary.
+Step 1/4 — Understand the dataset and produce a short summary.
 
 Requirements:
 - 8–15 concise bullets covering subjects/sessions, modalities (T1w/T2w/DWI/fMRI/etc.), tasks, naming patterns, id conventions.
@@ -47,25 +47,11 @@ Output:
 """
 
 
-def create_root_prompt(*, output_root: str, dataset_xml: Optional[str], readme_text: Optional[str], publication_text: Optional[str]) -> str:
-	return f"""
-Step 2/5 — Propose commands to create a new BIDS root directory.
-
-Constraints:
-- Use $OUTPUT_ROOT if present, otherwise use: {output_root}
-- Use mkdir -p; don't overwrite existing files.
-- Optionally create a minimal skeleton (.bidsignore, empty dirs if helpful).
-
-Context:\n{_ctx(dataset_xml, readme_text, publication_text)}
-
-Output:
-- A brief plan (2–5 bullets) followed by exactly one fenced bash block with commands only.
-"""
 
 
 def create_metadata_prompt(*, output_root: str, dataset_xml: Optional[str], readme_text: Optional[str], publication_text: Optional[str]) -> str:
 	return f"""
-Step 3/5 — Propose commands to create required BIDS metadata files.
+Step 2/4 — Propose commands to create required BIDS metadata files.
 
 Must include:
 - dataset_description.json (Name, BIDSVersion, License if known)
@@ -86,7 +72,7 @@ Output:
 
 def create_structure_prompt(*, output_root: str, dataset_xml: Optional[str], readme_text: Optional[str], publication_text: Optional[str]) -> str:
 	return f"""
-Step 4/5 — Propose commands to create the BIDS directory structure.
+Step 3/4 — Propose commands to create the BIDS directory structure.
 
 Goals:
 - Infer subjects, sessions, and modalities; create sub-<label>/, optional ses-<label>/, and modality folders (anat, dwi, func, fmap, etc.).
@@ -105,7 +91,7 @@ Output:
 
 def rename_and_move_prompt(*, output_root: str, dataset_xml: Optional[str], readme_text: Optional[str], publication_text: Optional[str]) -> str:
 	return f"""
-Step 5/5 — Propose commands to rename and move files into the BIDS structure.
+Step 4/4 — Propose commands to rename and move files into the BIDS structure.
 
 Requirements:
 - Map original names to BIDS filenames; demonstrate patterns (e.g., with find/xargs) carefully.