- save outputs into dbs

Project.py

@@ -1,5 +1,6 @@
 from typing import Any, Dict
 from common_functions_v4 import *
+from common_functions_v4 import log_prompt
 from page_prompts_config import PROMPTS, ModelType
 from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 import openai
@@ -55,6 +56,10 @@ class Project:
         self.component_list = []
         self.project_detail = [] 
         
+        self.plan_test_mandays_output_id = None
+        self.dev_mandays_output_id = None
+        self.mvp_mandays_output_id = None
+        
         # Initialize all prompt outputs as attributes
         for config in PROMPTS.values():
             for output in config.outputs:
@@ -76,6 +81,10 @@ class Project:
         'mvp_components': lambda self: self.mvp_components,
         'revised_mandays_estimates': lambda self: self.revised_mandays,
         'formatted_mvp_mandays': lambda self: self.formatted_mvp_mandays,
+        'generated_mvp_prd': lambda self: self.mvp_prd,
+        'general_sow': lambda self: self.general_sow,
+        'detailed_sow_json': lambda self: self.detailed_sow_json,
+        'mvp_mandays': lambda self: self.mvp_mandays,
     }
 
     def execute_prompt(self, prompt_name: str, input_variables: Dict[str, Any] = None) -> str:
@@ -180,16 +189,13 @@ class Project:
                 "project_detail": self.get_project_detail()
             }
         )
+        
 
     #####################################################################
     def generate_prd_and_components(self, progress=gr.Progress()):
         """Generate PRD and components from project details"""
         
         # Step 1 : Log & Generate PRD
-        try: 
-             log_prompt_execution(PROMPTS['generate_prd'].step, PROMPTS['generate_prd'].description, PROMPTS["generate_prd"].prompt)
-        except Exception as e:
-            print(f"Error logging PRD generation: {str(e)}")
         progress(0, desc="Progress 1: Generating PRD from Q&A...")
         self.generated_prd = self.execute_prompt(
             "generate_prd",
@@ -197,14 +203,9 @@ class Project:
                 "project_detail": self.get_project_detail()
             }
         )
-
+        log_prompt(PROMPTS['generate_prd'].step, PROMPTS['generate_prd'].description, PROMPTS["generate_prd"].prompt, self.generated_prd)
+        
         # Step 2 : Log & Generate Plan & Test Components
-        try:
-            log_prompt_execution(PROMPTS['generate_plan_test_components'].step, PROMPTS['generate_plan_test_components'].description, PROMPTS["generate_plan_test_components"].prompt)
-        except Exception as e:
-            print(f"Error logging PRD generation: {str(e)}")
-
-        # Step 2: Generate Components
         progress(0.4, desc="Progress 2: Generating Planning & Testing Component...")
         self.plan_test_component_list = self.execute_prompt(
             "generate_plan_test_components",
@@ -212,13 +213,9 @@ class Project:
                 "generated_prd": self.generated_prd
             }
         )
-
+        log_prompt(PROMPTS['generate_plan_test_components'].step, PROMPTS['generate_plan_test_components'].description, PROMPTS["generate_plan_test_components"].prompt, self.plan_test_component_list)
+        
         # Step 3 : Log & Generate Dev Components
-        try:
-            log_prompt_execution(PROMPTS['generate_dev_components'].step, PROMPTS['generate_dev_components'].description, PROMPTS["generate_dev_components"].prompt)
-        except Exception as e:
-            print(f"Error logging Plan Test Components generation: {str(e)}")
-
         progress(0.7, desc="Progress 3: Generating Development Components...")
         self.dev_component_list = self.execute_prompt(
             "generate_dev_components",
@@ -226,13 +223,9 @@ class Project:
                 "generated_prd": self.generated_prd
             }
         )
-
+        log_prompt(PROMPTS['generate_dev_components'].step, PROMPTS['generate_dev_components'].description, PROMPTS["generate_dev_components"].prompt, self.dev_component_list)
+        
         # Step 4 : Log & Reformat Dev Components
-        try:
-            log_prompt_execution(PROMPTS['reformat_dev_components'].step, PROMPTS['reformat_dev_components'].description, PROMPTS["reformat_dev_components"].prompt)
-        except Exception as e:
-            print(f"Error logging Development Components generation: {str(e)}")
-
         progress(0.8, desc="Progress 4: Reformatting Development Components...")
         self.reformatted_dev_component_list = self.execute_prompt(
             "reformat_dev_components",
@@ -240,6 +233,7 @@ class Project:
                 "generated_dev_components": self.dev_component_list
             }
         )
+        log_prompt(PROMPTS['reformat_dev_components'].step, PROMPTS['reformat_dev_components'].description, PROMPTS["reformat_dev_components"].prompt, self.reformatted_dev_component_list)
         
         progress(1.0, desc="Complete!")
         
@@ -263,11 +257,6 @@ class Project:
         self.reformatted_dev_components = dev_component
   
         # Step 1 : Log & Generate Plan & Test Mandays
-        try:
-          log_prompt_execution(PROMPTS['generate_plan_test_mandays'].step, PROMPTS['generate_plan_test_mandays'].sub_step, PROMPTS["generate_plan_test_mandays"].prompt)
-        except Exception as e:
-          print(f"Error logging Plan Test Mandays generation: {str(e)}")
-          
         progress(0.4, desc="Progress 1: Generating Mandays for Plan & Test Components...")
         self.plan_test_mandays = self.execute_prompt(
             "generate_plan_test_mandays",
@@ -275,13 +264,9 @@ class Project:
                 "generated_plan_test_components": plan_test_component
             }
         )
-  
+        self.plan_test_mandays_output_id = log_prompt(PROMPTS['generate_plan_test_mandays'].step, PROMPTS['generate_plan_test_mandays'].description, PROMPTS["generate_plan_test_mandays"].prompt, self.plan_test_mandays)
+        
         # Step 2 : Log & Generate Dev Mandays
-        try:
-          log_prompt_execution(PROMPTS['generate_dev_mandays'].step, PROMPTS['generate_dev_mandays'].sub_step, PROMPTS["generate_dev_mandays"].prompt)
-        except Exception as e:
-          print(f"Error logging Dev Mandays generation: {str(e)}")
-          
         progress(0.8, desc="Progress 2: Generating Mandays for Dev Components...")
         self.dev_mandays = self.execute_prompt(
             "generate_dev_mandays",
@@ -289,7 +274,8 @@ class Project:
                 "reformatted_dev_components": dev_component
             }
         )
-    
+        self.dev_mandays_output_id = log_prompt(PROMPTS['generate_dev_mandays'].step, PROMPTS['generate_dev_mandays'].description, PROMPTS["generate_dev_mandays"].prompt, self.dev_mandays)
+        
         progress(1.0, desc="Complete!")
         return [
             pd.read_csv(StringIO(self.plan_test_mandays), on_bad_lines='skip'),
@@ -301,11 +287,6 @@ class Project:
         """Step 2.2: Analyze components"""
         
         # Step 1 : Log & Analyze Plan & Test Mandays
-        try:
-          log_prompt_execution(PROMPTS['analyze_planning_testing_mandays'].step, PROMPTS['analyze_planning_testing_mandays'].sub_step, PROMPTS["analyze_planning_testing_mandays"].prompt)
-        except Exception as e:
-          print(f"Error logging Planning & Testing Components analysis: {str(e)}")
-          
         progress(0.4, desc="Progress 1: Analyzing Planning & Testing Components...")
         self.planning_testing_components = self.execute_prompt(
             "analyze_planning_testing_mandays",
@@ -313,13 +294,9 @@ class Project:
                 "generated_plan_test_mandays": self.plan_test_mandays
             }
         )
+        log_prompt(PROMPTS['analyze_planning_testing_mandays'].step, PROMPTS['analyze_planning_testing_mandays'].description, PROMPTS["analyze_planning_testing_mandays"].prompt, self.planning_testing_components)
         
         # Step 2 : Log & Analyze Dev Mandays
-        try:
-          log_prompt_execution(PROMPTS['analyze_development_mandays'].step, PROMPTS['analyze_development_mandays'].sub_step, PROMPTS["analyze_development_mandays"].prompt)
-        except Exception as e:
-          print(f"Error logging Development Components analysis: {str(e)}")
-    
         progress(0.8, desc="Progress 2: Analyzing Development Components...")
         self.development_components = self.execute_prompt(
             "analyze_development_mandays",
@@ -327,7 +304,8 @@ class Project:
                 "generated_dev_mandays": self.dev_mandays
             }
         )
-    
+        log_prompt(PROMPTS['analyze_development_mandays'].step, PROMPTS['analyze_development_mandays'].description, PROMPTS["analyze_development_mandays"].prompt, self.development_components)
+         
         # Combine the results from both analyses
         self.mvp_components = {
             "planning_testing": self.planning_testing_components,
@@ -341,11 +319,6 @@ class Project:
         """Step 2.3: Generate MVP estimates"""
         
         # Step 1 : Log & Recalculate Mandays
-        try:
-          log_prompt_execution(PROMPTS['recalculate_mandays'].step, PROMPTS['recalculate_mandays'].sub_step, PROMPTS["recalculate_mandays"].prompt)
-        except Exception as e:
-          print(f"Error logging MVP Mandays recalculation: {str(e)}")
-          
         progress(0.3, desc="Progress 1: Recalculating Mandays...")
         self.revised_mandays = self.execute_prompt(
             "recalculate_mandays",
@@ -354,32 +327,28 @@ class Project:
                 "generated_prd": self.generated_prd
             }
         )
+        log_prompt(PROMPTS['recalculate_mandays'].step, PROMPTS['recalculate_mandays'].description, PROMPTS["recalculate_mandays"].prompt, self.revised_mandays)
         
         # Step 2 : Log & Generate MVP Mandays
-        try:
-          log_prompt_execution(PROMPTS['generate_MVP_mandays'].step, PROMPTS['generate_MVP_mandays'].sub_step, PROMPTS["generate_MVP_mandays"].prompt)
-        except Exception as e:
-          print(f"Error logging MVP Mandays generation: {str(e)}")
-          
         progress(0.6, desc="Progress 2: Formatting MVP Mandays...")
-        mvp_mandays = self.execute_prompt(
+        self.mvp_mandays = self.execute_prompt(
             "generate_MVP_mandays",
             {
                 "revised_mandays_estimates": self.revised_mandays
             }
         )
-
+        
         try:
             # Process MVP mandays into dataframe
-            mvp_mandays = (mvp_mandays
+            self.mvp_mandays = (self.mvp_mandays
                 .replace("```csv\n", "")
                 .replace("```\n", "")
                 .replace("```", "")
                 .strip())
             
-            sections = (mvp_mandays.split("---SECTION BREAK---") 
-                       if "---SECTION BREAK---" in mvp_mandays 
-                       else mvp_mandays.split("\n---\n"))
+            sections = (self.mvp_mandays.split("---SECTION BREAK---") 
+                       if "---SECTION BREAK---" in self.mvp_mandays 
+                       else self.mvp_mandays.split("\n---\n"))
             
             if len(sections) != 2:
                 raise ValueError("MVP mandays output must contain exactly two sections")
@@ -411,7 +380,7 @@ class Project:
             for df in [plan_test_df, dev_df, mvp_df]:
                 df['mandays'] = pd.to_numeric(df['mandays'].replace('', '0'), errors='coerce').fillna(0)
 
-            total_mandays, total_cost, estimated_months, mvp_mandays, mvp_cost, mvp_estimated_months = calculate_mandays_and_costs(plan_test_df, dev_df, mvp_df)
+            total_mandays, total_cost, estimated_months, total_mvp_mandays, mvp_cost, mvp_estimated_months = calculate_mandays_and_costs(plan_test_df, dev_df, mvp_df)
             
             cost_summary = f"""
                 Original Estimate:
@@ -420,18 +389,19 @@ class Project:
                 ({estimated_months:.2}months)
                 
                 MVP-Based Estimate:
-                Total Mandays: {mvp_mandays:.2f}
+                Total Mandays: {total_mvp_mandays:.2f}
                 Total Cost: ${mvp_cost:,.2f}
                 ({mvp_estimated_months:.2}months)
                 """
-
+            self.mvp_mandays_output_id = log_prompt(PROMPTS['generate_MVP_mandays'].step, PROMPTS['generate_MVP_mandays'].description, PROMPTS["generate_MVP_mandays"].prompt, self.mvp_mandays)
+            
             progress(1.0, desc="Complete!")
             return [mvp_df, "Generated MVP estimates!", cost_summary]
 
         except Exception as e:
             return [None, f"Error processing MVP mandays: {str(e)}", "Error calculating costs"]
 
-   ################################################################################################
+    ################################################################################################
     def generate_sow(self, generated_prd, plan_test_component, dev_component, mvp_components , cost, progress=gr.Progress()):
         self.generated_plan_test_components = plan_test_component
         self.reformatted_dev_components = dev_component
@@ -439,28 +409,19 @@ class Project:
         self.mvp_components = mvp_components
 
         # Step 1 : Log & Generate MVP PRD
-        try:
-          log_prompt_execution(PROMPTS['generate_mvp_prd'].step, PROMPTS['generate_mvp_prd'].description, PROMPTS["generate_mvp_prd"].prompt)
-        except Exception as e:
-          print(f"Error logging MVP PRD generation: {str(e)}")
-          
         progress(0.2, desc="Progress 1: Drafting MVP PRD")
-        mvp_prd = self.execute_prompt(
+        self.mvp_prd = self.execute_prompt(
             "generate_mvp_prd",
             {
                 "generated_prd": generated_prd,
                 "mvp_components": mvp_components
             }
         )
+        log_prompt(PROMPTS['generate_mvp_prd'].step, PROMPTS['generate_mvp_prd'].description, PROMPTS["generate_mvp_prd"].prompt, self.mvp_prd)
         
         # Step 2 : Log & Generate Business SOW
-        try:
-          log_prompt_execution(PROMPTS['generate_BD_SOW'].step, PROMPTS['generate_BD_SOW'].description, PROMPTS["generate_BD_SOW"].prompt)
-        except Exception as e:
-          print(f"Error logging MVP PRD generation: {str(e)}")
-          
         progress(0.4, desc="Progress 2: Drafting SOW")
-        general_sow = self.execute_prompt(
+        self.general_sow = self.execute_prompt(
             "generate_BD_SOW",
             {
                 "generated_prd": generated_prd,
@@ -469,15 +430,11 @@ class Project:
                 "quotation_cost": cost
             }
         )
-
+        log_prompt(PROMPTS['generate_BD_SOW'].step, PROMPTS['generate_BD_SOW'].description, PROMPTS["generate_BD_SOW"].prompt, self.general_sow)
+        
         # Step 3 : Log & Generate Technical SOW
-        try:
-          log_prompt_execution(PROMPTS['generate_Tech_SOW'].step, PROMPTS['generate_Tech_SOW'].description, PROMPTS["generate_Tech_SOW"].prompt)
-        except Exception as e:
-          print(f"Error logging Technical SOW generation: {str(e)}")
-          
         progress(0.8, desc="Progress 3: Drafting Technical SOW")
-        detailed_sow_json = self.execute_prompt(
+        self.detailed_sow_json = self.execute_prompt(
             "generate_Tech_SOW",
             {
                 "generated_plan_test_components": plan_test_component,
@@ -485,23 +442,24 @@ class Project:
                 "mvp_components": mvp_components,
             }
         )
-
+        log_prompt(PROMPTS['generate_Tech_SOW'].step, PROMPTS['generate_Tech_SOW'].description, PROMPTS["generate_Tech_SOW"].prompt, self.detailed_sow_json)
+         
         try:
             # Parse detailed_sow into a JSON object
-            detailed_sow_json = json.loads(detailed_sow_json)
+            self.detailed_sow_json = json.loads(self.detailed_sow_json)
 
             # Extract required fields
-            scope_summary = detailed_sow_json.get("scope_summary", "")
-            modules_and_functional_requirements = detailed_sow_json.get("modules_and_functional_requirements", "")
-            out_of_scope = detailed_sow_json.get("out_of_scope", "")
-            system_flow = detailed_sow_json.get("system_flow", "")
+            scope_summary = self.detailed_sow_json.get("scope_summary", "")
+            modules_and_functional_requirements = self.detailed_sow_json.get("modules_and_functional_requirements", "")
+            out_of_scope = self.detailed_sow_json.get("out_of_scope", "")
+            system_flow = self.detailed_sow_json.get("system_flow", "")
 
             # Combine all fields into detailed_sow
             detailed_sow = f"{scope_summary}\n\n{modules_and_functional_requirements}\n\n{out_of_scope}\n\n{system_flow}"
 
             # Create final SOW
-            final_general_sow = f"**Hi , some sections of this SOW is generated seprately from the main draft. youll have to move this to the right spot manually ;)\n **Project Quotation:**\n{cost}\n{modules_and_functional_requirements}\n\n {general_sow}\nEOF"
-            return [mvp_prd , mvp_prd ,final_general_sow, final_general_sow, detailed_sow, detailed_sow, "Generated SOW!"]
+            final_general_sow = f"**Hi , some sections of this SOW is generated seprately from the main draft. youll have to move this to the right spot manually ;)\n **Project Quotation:**\n{cost}\n{modules_and_functional_requirements}\n\n {self.general_sow}\nEOF"
+            return [self.mvp_prd , self.mvp_prd ,final_general_sow, final_general_sow, detailed_sow, detailed_sow, "Generated SOW!"]
         except Exception as e:
             return ["Error generating SOW", "Error: " + str(e), "Failed to generate SOW"]
 
@@ -515,7 +473,11 @@ class Project:
                 mvp_df['mandays'] = pd.to_numeric(mvp_df['mandays'].replace('', '0'), errors='coerce').fillna(0)
         
             # Calculate totals
-            total_mandays, total_cost, estimated_months, mvp_mandays, mvp_cost, mvp_estimated_months = calculate_mandays_and_costs(plan_test_df, dev_df, mvp_df)
+            total_mandays, total_cost, estimated_months, total_mvp_mandays, mvp_cost, mvp_estimated_months = calculate_mandays_and_costs(plan_test_df, dev_df, mvp_df)
+            
+            update_prompt_execution_output(self.plan_test_mandays_output_id , plan_test_df)
+            update_prompt_execution_output(self.dev_mandays_output_id , dev_df)
+            update_prompt_execution_output(self.mvp_mandays_output_id , mvp_df)
 
             status_message = f"Successfully Updated Quotation. SessionID:{self.session_id}"
             cost_summary = f"""
@@ -525,7 +487,7 @@ class Project:
                 ({estimated_months:.2}months)
                 
                 MVP-Based Estimate:
-                Total Mandays: {mvp_mandays:.2f}
+                Total Mandays: {total_mvp_mandays:.2f}
                 Total Cost: ${mvp_cost:,.2f}
                 ({mvp_estimated_months:.2}months)"""
             

app.py

@@ -259,7 +259,7 @@ def create_quotation_generator_section():
                                 
                                 with gr.Column(scale=1):
                                     quotation_cost = gr.Textbox(label="Cost Summary", lines=3, interactive=False)
-                                    page_recalc_btn = gr.Button("Recalculate")
+                                    page_recalc_btn = gr.Button("Recalculate Cost")
                                     page_notes_box = gr.Textbox(
                                         label="Notes", 
                                         lines=3, 

common_functions_v4.py

@@ -1,4 +1,5 @@
 import uuid
+import numpy as np
 import pandas as pd
 from io import StringIO
 # from pathlib import Path
@@ -327,7 +328,7 @@ def fetch_session(session_id):
         # return "", "", f"Error fetching session: {str(e)}", "", ""
 
 def log_prompt_execution(step_name, sub_step_name, prompt_text):
-    """Log prompt execution to the database with a randomly generated prompt_id, avoiding duplicates for the same prompt text."""
+    """Log prompt execution to the database with a randomly generated prompt_id."""
     created_at = datetime.datetime.now()
     session_id = state.quotation_project.session_id
     
@@ -340,27 +341,82 @@ def log_prompt_execution(step_name, sub_step_name, prompt_text):
         conn = get_db_connection()
         cur = conn.cursor()
         
-        # Check for existing record with the same prompt text
+        # Directly insert the prompt execution without checking for duplicates
         cur.execute("""
-            SELECT COUNT(*) FROM prompts 
-            WHERE session_id = %s AND step_name = %s AND sub_step_name = %s AND prompt = %s
-        """, (session_id, step_name, sub_step_name, prompt_text))
+            INSERT INTO prompts (session_id, prompt_id, step_name, sub_step_name, prompt, created_at)
+            VALUES (%s, %s, %s, %s, %s, %s)
+        """, (session_id, prompt_id, step_name, sub_step_name, prompt_text, created_at))
+        conn.commit()
+        return prompt_id  
+    except Exception as e:
+        print(f"Error logging prompt execution: {str(e)}")
+        return None  
+    finally:
+        if conn:
+            conn.close() 
+            
+def log_prompt_execution_output(prompt_id, output):
+    """Save prompt execution output to the database"""
+    
+    created_at = datetime.datetime.now()
+    output_id = str(uuid.uuid4())
+    try:
+        # Establish a database connection
+        conn = get_db_connection()
+        cur = conn.cursor()
+
+        # Directly insert the output into the outputs table without checking for duplicates
+        cur.execute("""
+            INSERT INTO outputs (output_id, prompt_id, output, created_at)
+            VALUES (%s, %s, %s, %s)
+        """, (output_id, prompt_id, output, created_at))
+
+        # Commit the transaction
+        conn.commit()
         
-        count = cur.fetchone()[0]
+    except Exception as e:
+        print(f"Error logging prompt execution output: {str(e)}")
+        return None  
+    finally:
+        if conn:
+            conn.close() 
+            
+def update_prompt_execution_output(output_id, output):
+    """Update prompt execution output in the database based on output_id"""
+    
+    created_at = datetime.datetime.now()
+    try:
+        # Establish a database connection
+        conn = get_db_connection()
+        cur = conn.cursor()
+        
+         # Convert output to a standard Python type if it's a numpy type
+        if isinstance(output, (np.integer, float)):  
+            output = int(output) if isinstance(output, np.integer) else float(output)
+
+        # Update the existing output for the given output_id
+        cur.execute("""
+            UPDATE outputs
+            SET output = %s, created_at = %s
+            WHERE output_id = %s
+        """, (output, created_at, output_id))
+
+        # Commit the transaction
+        conn.commit()
         
-        if count == 0:  # Only insert if no existing record with the same prompt text is found
-            cur.execute("""
-                INSERT INTO prompts (session_id, prompt_id, step_name, sub_step_name, prompt, created_at)
-                VALUES (%s, %s, %s, %s, %s, %s)
-            """, (session_id, prompt_id, step_name, sub_step_name, prompt_text, created_at))
-            conn.commit()
-            return prompt_id  
-        else:
-            print("Duplicate prompt text found for the same session and step. Skipping insertion.")
-            return None 
     except Exception as e:
-        print(f"Error logging prompt execution: {str(e)}")
+        print(f"Error updating prompt execution output: {str(e)}")
         return None  
     finally:
         if conn:
-            conn.close() 
+            conn.close()
+            
+def log_prompt(prompt_name: str, prompt_description: str, prompt: str, output: Any) -> None:
+        """Log prompt execution and its output"""
+        try:
+            prompt_id = log_prompt_execution(prompt_name, prompt_description, prompt)
+            output_id = log_prompt_execution_output(prompt_id, output)
+            return output_id
+        except Exception as e:
+            print(f"Error logging {prompt_name} generation: {str(e)}")
+            

page_prompts_config.py

@@ -505,7 +505,7 @@ PROMPTS = {
         inputs=["generated_plan_test_components"],
         outputs=["generated_plan_test_mandays"],
         model=ModelType.O1_MINI,
-        description="Generate planning and testing mandays",
+        description="Step 2.1 : Generate planning and testing mandays",
         step="Step 2 : Mandays & Quotation",
         sub_step="Step 2.1 : Generate Mandays",
         ui={
@@ -564,7 +564,7 @@ PROMPTS = {
         inputs=["reformatted_dev_components"],
         outputs=["generated_dev_mandays"],
         model=ModelType.O1_MINI,
-        description="Generate development mandays",
+        description="Step 2.1 : Generate development mandays",
         step="Step 2 : Mandays & Quotation",
         sub_step="Step 2.1 : Generate Mandays",
         ui={
@@ -776,7 +776,7 @@ PROMPTS = {
         inputs=["generated_plan_test_components", "reformatted_dev_components" ,"mvp_components"],
         outputs=["generated_Tech_SOW"],
         model=ModelType.O1_MINI,
-        description="Step 3.3 : Generate Test SOW",
+        description="Step 3.3 : Generate Tech SOW",
         step="Step 3 : Final Documentation",
         ui={
             "Tech_SOW_prompt_editor": UIConfig(
@@ -821,7 +821,7 @@ PROMPTS = {
         inputs=["generated_plan_test_mandays"],
         outputs=["identified_planning_testing_components"],
         model=ModelType.O1_MINI,
-        description="Analyze planning and testing components",
+        description="Step 2.2 : Analyze planning and testing components",
         step="Step 2 : Mandays & Quotation",
         sub_step="Step 2.2 : Analyze Components",
         ui={
@@ -862,7 +862,7 @@ PROMPTS = {
         inputs=["generated_dev_mandays"],
         outputs=["identified_development_components"],
         model=ModelType.O1_MINI,
-        description="Analyze development components",
+        description="Step 2.2 : Analyze development components",
         step="Step 2 : Mandays & Quotation",
         sub_step="Step 2.2 : Analyze Components",
         ui={
@@ -912,7 +912,7 @@ PROMPTS = {
         inputs=["identified_priority_components", "generated_prd"],
         outputs=["revised_mandays_estimates"],
         model=ModelType.O1_MINI,
-        description="Recalculate Mandays",
+        description="Step 2.3 : Recalculate Mandays",
         step="Step 2 : Mandays & Quotation",
         sub_step="Step 2.3 : Recalculate MVP Mandays",
         ui={
@@ -955,7 +955,7 @@ PROMPTS = {
         inputs=["revised_mandays_estimates"],
         outputs=["generated_MVP_mandays"],
         model=ModelType.O1_MINI,
-        description="Generate MVP Mandays",
+        description="Step 2.3 : Generate MVP Mandays",
         step="Step 2 : Mandays & Quotation",
         sub_step="Step 2.3 : Recalculate MVP Mandays",
         ui={