add: question agent partially functional

Project.py

@@ -265,6 +265,10 @@ class Project:
             for output in config.outputs:
                 setattr(self, output, "")
 
+    async def generate_client_initial_question(self):
+        """Generate follow-up questions after initial client response"""
+        return PROMPTS["client_initial_question"].prompt
+    
     async def generate_client_follow_up(self):
         """Generate follow-up questions after initial client response"""
         return await self.async_execute_prompt(

common_functions_v4.py

@@ -277,7 +277,7 @@ async def clean_sample_answers(text):
             r'(?i)[\(\[]?\s*Sample\s*:\s*"[^"]*"\s*[\)\]]?',
             'Answer:',
             text,
-            flags=re.MULTILINE
+            flags=re.MULTILINE | re.IGNORECASE
         )
         
         return cleaned_text
@@ -288,19 +288,7 @@ async def clean_sample_answers(text):
 # Example usage:
 # chat_history = [{'role': 'user', ...}, {'role': 'assistant', ...}]
 # log_chat_history(chat_history)    
-# if len(history) == 1:  # After first client information question
-#             next_question = state.quotation_project.generate_client_follow_up()
-#         elif len(history) == 2: 
-#             # Get configuration output
-#             next_question = validate_project_choice(answer,state.quotation_project)
-#             # next_question = result
-#         elif len(history) == 3:  # After specific questioning, run general questions
-#             next_question = state.quotation_project.generate_general_questions()
-        
-#         else:  # Subsequent project requirements questions
-#             # next_question = quotation_project.generate_follow_up()
-#             next_question = state.quotation_project.generate_further_follow_up_questions()
-        
+#
 #TODO: Ensure it directs towards correct question
 async def async_process_response(answer, history):
     """Process user responses and generate appropriate follow-up questions."""
@@ -330,10 +318,32 @@ async def async_process_response(answer, history):
             metadata={"title": "_Processing_ step-by-step", "id": 0, "status": "pending"}
         )
         yield response, ""
-    
+
+        function_to_run = state.quotation_project.generate_client_initial_question
         current_thoughts = PROMPTS["client_initial_question"].thoughts
+
+        project_detail_len = len(state.quotation_project.project_detail)
         
-        next_question_task = asyncio.create_task(state.quotation_project.generate_client_follow_up())
+        if project_detail_len == 2:
+            function_to_run = state.quotation_project.generate_client_follow_up
+            current_thoughts = PROMPTS["generate_client_follow_up"].thoughts
+
+        elif project_detail_len == 3:
+            function_to_run = lambda: run_question_agent(state.quotation_project)
+            current_thoughts = PROMPTS["generate_client_follow_up"].thoughts
+
+        elif project_detail_len == 4:
+            function_to_run = state.quotation_project.generate_general_questions
+            current_thoughts = PROMPTS["generate_general_questions"].thoughts
+
+        elif project_detail_len >= 5:
+            function_to_run = state.quotation_project.generate_further_follow_up_questions
+            current_thoughts = PROMPTS["generate_further_follow_up_questions"].thoughts
+
+
+        
+        
+        next_question_task = asyncio.create_task(function_to_run())
 
         # Update project detail with new list;
         if state.quotation_project.session_id:
@@ -436,30 +446,30 @@ def calculate_mvp_mandays_and_costs(generated_mvp_results):
 def create_new_session():
     """Create a new session in the database and return the session_id"""
     try:
-        # conn = get_db_connection()
-        # cur = conn.cursor()
-        
-        # # Insert new session with start time
-        # cur.execute("""
-        #     INSERT INTO sessions (start_time)
-        #     VALUES (CURRENT_TIMESTAMP)
-        #     RETURNING session_id
-        # """)
-        
-        # session_id = cur.fetchone()[0]
-        
-        # # Insert session_base_project record for "Page"
-        # cur.execute("""
-        #     INSERT INTO session_base_project (session_id, base_project_name)
-        #     VALUES (%s, 'Page')
-        # """, (session_id,))
-        
-        # conn.commit()
-        # cur.close()
-        # conn.close()
-        
-        # return session_id
-        return 151
+        conn = get_db_connection()
+        cur = conn.cursor()
+        
+        # Insert new session with start time
+        cur.execute("""
+            INSERT INTO sessions (start_time)
+            VALUES (CURRENT_TIMESTAMP)
+            RETURNING session_id
+        """)
+        
+        session_id = cur.fetchone()[0]
+        
+        # Insert session_base_project record for "Page"
+        cur.execute("""
+            INSERT INTO session_base_project (session_id, base_project_name)
+            VALUES (%s, 'Page')
+        """, (session_id,))
+        
+        conn.commit()
+        cur.close()
+        conn.close()
+        
+        return session_id
+        # return 151
     except Exception as e:
         print(f"Error creating new session: {str(e)}")
         return None

prompt_configs.py

@@ -142,6 +142,16 @@ PROMPTS = {
                 show_copy_button=True
             )
         },
+        thoughts=[
+            "Read the project detail and function list.",
+            "Identify the project requirements and function list.",
+            "Determine the project configuration type.",
+            "Output a JSON object with the configuration type and selected functions.",
+            "Running the selected functions to generate questions.",
+            "Analyzing the project requirements to determine the best question set.",
+            "Generating context-aware questions based on the project type.",
+            "Formatting the output with sample answers for clarity."
+        ],
     ),
 
     "component_agent": PromptConfig(