Feat/24 more verbose analyze process

src/repello_agent_wiz/cli.py

@@ -1,33 +1,73 @@
 import argparse
-import os  
-import sys 
+from tqdm import tqdm
+import time
+import threading
 from .frameworks import agent_chat, autogen, crewai, google_adk, langgraph, llama_index, n8n, openai_agents, pydantic, swarm
 from .analyzers import generate_maestro_analysis_report
 from .visualizers.visualizer import generate_visualization
 
 
+def run_with_timer(task_fn, *args, label="Working", **kwargs):
+    stop_event = threading.Event()
+    total = 100
+    pbar = tqdm(total=total, desc=f"⏳ {label}",
+                bar_format="{desc} |{bar}| {percentage:3.0f}% | {elapsed}s")
+
+    def animate():
+        while not stop_event.is_set():
+            if pbar.n < total:
+                pbar.update(1)
+            else:
+                pbar.n = 0
+                pbar.refresh()
+            time.sleep(0.1)
+
+    t = threading.Thread(target=animate)
+    t.start()
+
+    try:
+        result = task_fn(*args, **kwargs)
+    finally:
+        stop_event.set()
+        t.join()
+        pbar.n = total
+        pbar.refresh()
+        pbar.close()
+        elapsed = time.time() - (time.time() - pbar.last_print_t)
+        print(f"✅ {label} completed in {elapsed:.1f}s")
+    return result
+
+
 def main():
-    parser = argparse.ArgumentParser(description="Agent Wiz CLI: Extract, Analyze, Visualize agentic workflows.")
+    parser = argparse.ArgumentParser(
+        description="Agent Wiz CLI: Extract, Analyze, Visualize agentic workflows.")
     subparsers = parser.add_subparsers(dest="command", required=True)
 
     # --- Extract command ---
-    extract_parser = subparsers.add_parser("extract", help="Extract graph from source code")
+    extract_parser = subparsers.add_parser(
+        "extract", help="Extract graph from source code")
     extract_parser.add_argument("--framework", "-f", required=True, choices=[
-        "agent_chat", "autogen", "crewai", "google_adk" , "langgraph", 
+        "agent_chat", "autogen", "crewai", "google_adk", "langgraph",
         "llama_index", "n8n", "openai_agents", "pydantic", "swarm"
     ])
-    extract_parser.add_argument("--directory", "-d", default=".", help="Directory containing source code")
-    extract_parser.add_argument("--output", "-o", default="graph.json", help="Path to output JSON file")
+    extract_parser.add_argument(
+        "--directory", "-d", default=".", help="Directory containing source code")
+    extract_parser.add_argument(
+        "--output", "-o", default="graph.json", help="Path to output JSON file")
 
     # --- Analyze command ---
-    analyze_parser = subparsers.add_parser("analyze", help="Run threat modeling analysis on extracted graph")
-    analyze_parser.add_argument("--input", "-i", required=True, help="Path to JSON graph file")
+    analyze_parser = subparsers.add_parser(
+        "analyze", help="Run threat modeling analysis on extracted graph")
+    analyze_parser.add_argument(
+        "--input", "-i", required=True, help="Path to JSON graph file")
 
     # --- Visualize command ---
-    visualize_parser = subparsers.add_parser("visualize", help="Generate HTML visualization from graph JSON")
-    visualize_parser.add_argument("--input", "-i", required=True, help="Path to JSON graph file")
-    visualize_parser.add_argument("--open", action="store_true", help="Automatically open the HTML in a browser")
-
+    visualize_parser = subparsers.add_parser(
+        "visualize", help="Generate HTML visualization from graph JSON")
+    visualize_parser.add_argument(
+        "--input", "-i", required=True, help="Path to JSON graph file")
+    visualize_parser.add_argument(
+        "--open", action="store_true", help="Automatically open the HTML in a browser")
 
     # --- Parse and execute ---
     args = parser.parse_args()
@@ -48,31 +88,37 @@ def main():
 
             match args.framework:
                 case "agent_chat":
-                    agent_chat.extract_agentchat_graph(args.directory, args.output)
+                    agent_chat.extract_agentchat_graph(
+                        args.directory, args.output)
                 case "autogen":
                     autogen.extract_autogen_graph(args.directory, args.output)
                 case "crewai":
                     crewai.extract_crewai_graph(args.directory, args.output)
-                case "google_adk" :
+                case "google_adk":
                     google_adk.extract_adk_graph(args.directory, args.output)
                 case "langgraph":
-                    langgraph.extract_langgraph_graph(args.directory, args.output)
+                    langgraph.extract_langgraph_graph(
+                        args.directory, args.output)
                 case "llama_index":
-                    llama_index.extract_llamaindex_graph(args.directory, args.output)
+                    llama_index.extract_llamaindex_graph(
+                        args.directory, args.output)
                 case "pydantic":
-                    pydantic.extract_pydantic_ai_graph(args.directory, args.output)
+                    pydantic.extract_pydantic_ai_graph(
+                        args.directory, args.output)
                 case "swarm":
                     swarm.extract_swarm_graph(args.directory, args.output)
                 case "n8n":
                     n8n.extract_n8n_graph(args.directory, args.output)
                 case "openai_agents":
-                    openai_agents.extract_openai_agents_graph(args.directory, args.output)
+                    openai_agents.extract_openai_agents_graph(
+                        args.directory, args.output)
                 case _:
                     print(f"Unknown framework: {args.framework}")
-    
+
         case "analyze":
-            generate_maestro_analysis_report(args.input)
-
+            run_with_timer(generate_maestro_analysis_report,
+                           args.input, label="Generating threat model report")
+
         case "visualize":
             generate_visualization(args.input, open_browser=args.open)
         case _: