this thing damn works

app.log

@@ -50301,3 +50301,131 @@ For troubleshooting, visit: https://python.langchain.com/docs/troubleshooting/er
 2025-06-06 18:34:27:__main__:INFO: Starting the interface
 2025-06-06 18:34:31:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
 2025-06-06 18:35:28:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 18:45:50:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 18:52:40:__main__:INFO: Starting the interface
+2025-06-06 18:52:45:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 18:52:48:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 18:53:33:asyncio:ERROR: Task exception was never retrieved
+future: <Task finished name='Task-1197' coro=<<async_generator_athrow without __name__>()> exception=RuntimeError("generator didn't stop after athrow()")>
+Traceback (most recent call last):
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/__init__.py", line 2663, in astream
+    yield o
+GeneratorExit
+
+During handling of the above exception, another exception occurred:
+
+Traceback (most recent call last):
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/__init__.py", line 2596, in astream
+    async with AsyncPregelLoop(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/loop.py", line 1393, in __aexit__
+    return await exit_task
+  File "/usr/local/lib/python3.10/contextlib.py", line 714, in __aexit__
+    raise exc_details[1]
+  File "/usr/local/lib/python3.10/contextlib.py", line 697, in __aexit__
+    cb_suppress = await cb(*exc_details)
+  File "/usr/local/lib/python3.10/contextlib.py", line 249, in __aexit__
+    raise RuntimeError("generator didn't stop after athrow()")
+RuntimeError: generator didn't stop after athrow()
+2025-06-06 18:56:39:__main__:INFO: Starting the interface
+2025-06-06 18:56:43:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 18:58:35:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 18:58:35:__main__:ERROR: Exception occurred
+Traceback (most recent call last):
+  File "/home/user/app/app.py", line 97, in chat_fn
+    async for stream_mode, chunk in graph.astream(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/__init__.py", line 2655, in astream
+    async for _ in runner.atick(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/runner.py", line 400, in atick
+    _panic_or_proceed(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/runner.py", line 509, in _panic_or_proceed
+    raise exc
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/retry.py", line 136, in arun_with_retry
+    return await task.proc.ainvoke(task.input, config)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 676, in ainvoke
+    input = await step.ainvoke(input, config, **kwargs)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 440, in ainvoke
+    ret = await self.afunc(*args, **kwargs)
+  File "/home/user/app/graph.py", line 154, in guidance_node
+    print(f"Tools Enabled: {state.tools_enabled}")
+  File "/usr/local/lib/python3.10/site-packages/pydantic/main.py", line 891, in __getattr__
+    raise AttributeError(f'{type(self).__name__!r} object has no attribute {item!r}')
+AttributeError: 'GraphProcessingState' object has no attribute 'tools_enabled'
+2025-06-06 19:03:03:__main__:INFO: Starting the interface
+2025-06-06 19:03:07:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 19:03:17:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 19:03:21:__main__:ERROR: Exception occurred
+Traceback (most recent call last):
+  File "/home/user/app/app.py", line 97, in chat_fn
+    async for stream_mode, chunk in graph.astream(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/__init__.py", line 2655, in astream
+    async for _ in runner.atick(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/runner.py", line 400, in atick
+    _panic_or_proceed(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/runner.py", line 509, in _panic_or_proceed
+    raise exc
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/retry.py", line 136, in arun_with_retry
+    return await task.proc.ainvoke(task.input, config)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 678, in ainvoke
+    input = await step.ainvoke(input, config)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/utils/runnable.py", line 440, in ainvoke
+    ret = await self.afunc(*args, **kwargs)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/graph/branch.py", line 185, in _aroute
+    value = reader(config)
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/read.py", line 110, in do_read
+    return mapper(read(select, fresh))
+  File "/usr/local/lib/python3.10/site-packages/langgraph/graph/state.py", line 1050, in _coerce_state
+    return schema(**input)
+  File "/usr/local/lib/python3.10/site-packages/pydantic/main.py", line 214, in __init__
+    validated_self = self.__pydantic_validator__.validate_python(data, self_instance=self)
+pydantic_core._pydantic_core.ValidationError: 1 validation error for GraphProcessingState
+next_stage
+  Input should be a valid string [type=string_type, input_value=None, input_type=NoneType]
+    For further information visit https://errors.pydantic.dev/2.10/v/string_type
+2025-06-06 19:05:35:__main__:INFO: Starting the interface
+2025-06-06 19:05:39:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 19:05:59:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 19:06:36:asyncio:ERROR: Task exception was never retrieved
+future: <Task finished name='Task-1329' coro=<<async_generator_athrow without __name__>()> exception=RuntimeError("generator didn't stop after athrow()")>
+Traceback (most recent call last):
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/__init__.py", line 2663, in astream
+    yield o
+GeneratorExit
+
+During handling of the above exception, another exception occurred:
+
+Traceback (most recent call last):
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/__init__.py", line 2596, in astream
+    async with AsyncPregelLoop(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/loop.py", line 1393, in __aexit__
+    return await exit_task
+  File "/usr/local/lib/python3.10/contextlib.py", line 714, in __aexit__
+    raise exc_details[1]
+  File "/usr/local/lib/python3.10/contextlib.py", line 697, in __aexit__
+    cb_suppress = await cb(*exc_details)
+  File "/usr/local/lib/python3.10/contextlib.py", line 249, in __aexit__
+    raise RuntimeError("generator didn't stop after athrow()")
+RuntimeError: generator didn't stop after athrow()
+2025-06-06 19:14:45:__main__:INFO: Starting the interface
+2025-06-06 19:14:49:__main__:INFO: Greeting added for new user via handle_initial_greeting_load.
+2025-06-06 19:14:55:__main__:INFO: Prompt: You are a helpful assistant.
+2025-06-06 19:15:37:asyncio:ERROR: Task exception was never retrieved
+future: <Task finished name='Task-1216' coro=<<async_generator_athrow without __name__>()> exception=RuntimeError("generator didn't stop after athrow()")>
+Traceback (most recent call last):
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/__init__.py", line 2663, in astream
+    yield o
+GeneratorExit
+
+During handling of the above exception, another exception occurred:
+
+Traceback (most recent call last):
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/__init__.py", line 2596, in astream
+    async with AsyncPregelLoop(
+  File "/usr/local/lib/python3.10/site-packages/langgraph/pregel/loop.py", line 1393, in __aexit__
+    return await exit_task
+  File "/usr/local/lib/python3.10/contextlib.py", line 714, in __aexit__
+    raise exc_details[1]
+  File "/usr/local/lib/python3.10/contextlib.py", line 697, in __aexit__
+    cb_suppress = await cb(*exc_details)
+  File "/usr/local/lib/python3.10/contextlib.py", line 249, in __aexit__
+    raise RuntimeError("generator didn't stop after athrow()")
+RuntimeError: generator didn't stop after athrow()

app.py

@@ -125,8 +125,8 @@ async def chat_fn(user_input: str, history: dict, input_graph_state: dict, uuid:
                             input_graph_state["awaiting_human_input"] = True
                             input_graph_state["human_assistance_tool_id"] = msg_tool_call["id"]
 
-                            # Show input textbox to user and pause graph execution
-                            yield "\n".join(waiting_output_seq), gr.Textbox(visible=True, label="Your input"), gr.skip()
+                            # Indicate that human input is needed
+                            yield "\n".join(waiting_output_seq), input_graph_state, gr.skip(), True
                             return  # Pause execution, resume in next call
 
                         else:

graph.py

@@ -128,19 +128,14 @@ class GraphProcessingState(BaseModel):
 
 
 async def guidance_node(state: GraphProcessingState, config=None):
-    # Use logger.debug() for verbose debugging information
-    # This allows you to control its visibility via logging configuration
-    print("\n--- Guidance Node (Debug via print) ---") # Added a newline for clarity
-
+    print("\n--- Guidance Node (Debug via print) ---")
 
     print(f"Prompt: {state.prompt}")
     for message in state.messages:
         if isinstance(message, HumanMessage):
             print(f"Human: {message.content}")
         elif isinstance(message, AIMessage):
-            # Check if content is non-empty
             if message.content:
-                # If content is a list (e.g., list of dicts), extract text
                 if isinstance(message.content, list):
                     texts = [item.get('text', '') for item in message.content if isinstance(item, dict) and 'text' in item]
                     if texts:
@@ -156,21 +151,17 @@ async def guidance_node(state: GraphProcessingState, config=None):
     print(f"Search Enabled: {state.search_enabled}")
     print(f"Next Stage: {state.next_stage}")
 
-
-    # Log boolean completion flags
     print(f"Idea Complete: {state.idea_complete}")
     print(f"Brainstorming Complete: {state.brainstorming_complete}")
     print(f"Planning Complete: {state.planning_complete}")
     print(f"Drawing Complete: {state.drawing_complete}")
     print(f"Product Searching Complete: {state.product_searching_complete}")
     print(f"Purchasing Complete: {state.purchasing_complete}")
-    
-    
+
     guidance_node.count = getattr(guidance_node, 'count', 0) + 1
     print('\nGuidance Node called count', guidance_node.count)
-    print("\n--- End Guidance Node Debug ---") # Added for clarity
+    print("\n--- End Guidance Node Debug ---")
 
-    # Prepare context: stage completion statuses
     stage_order = ["brainstorming", "planning", "drawing", "product_searching", "purchasing"]
     completed = [stage for stage in stage_order if getattr(state, f"{stage}_complete", False)]
     incomplete = [stage for stage in stage_order if not getattr(state, f"{stage}_complete", False)]
@@ -180,17 +171,15 @@ async def guidance_node(state: GraphProcessingState, config=None):
     guidance_prompt = (
         "You are the guiding assistant. Based on the user's input and the current project status, "
         "help decide the next stage in the DIY process. Ask the user which stage to continue next if needed.\n\n"
-        f"CURRENT STATUS:\n{status_summary}\n\n"  # <-- The information was moved here
+        f"CURRENT STATUS:\n{status_summary}\n\n"
         "Available stages: brainstorming, planning, drawing, product_searching, purchasing."
     )
 
-    # Build final prompt with base and current prompt
     if state.prompt:
         final_prompt = "\n".join([guidance_prompt, state.prompt, ASSISTANT_SYSTEM_PROMPT_BASE])
     else:
         final_prompt = "\n".join([guidance_prompt, ASSISTANT_SYSTEM_PROMPT_BASE])
 
-    # Create prompt template
     prompt = ChatPromptTemplate.from_messages(
         [
             ("system", final_prompt),
@@ -198,45 +187,72 @@ async def guidance_node(state: GraphProcessingState, config=None):
         ]
     )
 
-    # Bind tools if needed (minimal here, mostly guidance)
-    assistant_model = model.bind_tools([])
+    assistant_model = model.bind_tools([human_assistance])
 
-    # Create the chain
     chain = prompt | assistant_model
 
-    
-
-    # Get response from assistant
-    response = await chain.ainvoke({"messages": state.messages}, config=config)
+    try:
+        response = await chain.ainvoke({"messages": state.messages}, config=config)
+
+        for msg in response:
+            if isinstance(msg, HumanMessage):
+                print("Human:", msg.content)
+            elif isinstance(msg, AIMessage):
+                if isinstance(msg.content, list):
+                    ai_texts = [part.get("text", "") for part in msg.content if isinstance(part, dict)]
+                    print("AI:", " ".join(ai_texts))
+                else:
+                    print("AI:", msg.content)
+
+        # Check for tool calls in the response
+        if hasattr(response, "tool_calls"):
+            for tool_call in response.tool_calls:
+                tool_name = tool_call['name']
+                if tool_name == "human_assistance":
+                    query = tool_call['args']['query']
+                    print(f"Human input needed: {query}")
+                    # Handle human assistance tool call
+                    # You can pause execution and wait for user input here
+
+        return {
+            "messages": [response],
+            "next_stage": incomplete[0] if incomplete else "brainstorming"
+        }
+    except Exception as e:
+        print(f"Error in guidance node: {e}")
+        return {
+            "messages": [AIMessage(content="Error in guidance node.")],
+            "next_stage": "brainstorming"
+        }
 
-    for msg in response:
-        if isinstance(msg, HumanMessage):
-            print("Human:", msg.content)
-        elif isinstance(msg, AIMessage):
-            if isinstance(msg.content, list):
-                ai_texts = [part.get("text", "") for part in msg.content if isinstance(part, dict)]
-                print("AI:", " ".join(ai_texts))
-            else:
-                print("AI:", msg.content)
+async def brainstorming_node(state: GraphProcessingState, config=None):
+    print("\n--- Guidance Node (Debug via print) ---") # Added a newline for clarity
 
-    return {
-        "messages": response,
-        "next_stage": incomplete[0] if incomplete else None
-    }
 
-async def brainstorming_node(state: GraphProcessingState, config=None):
-    print("\n--- brainstorming_node (Debug via print) ---")
     print(f"Prompt: {state.prompt}")
-
-    # Log the content of each message
     for message in state.messages:
-        if isinstance(message, (HumanMessage, AIMessage, SystemMessage, ToolMessage)):
-            print(f"Message Content: {message.content}")
+        if isinstance(message, HumanMessage):
+            print(f"Human: {message.content}")
+        elif isinstance(message, AIMessage):
+            # Check if content is non-empty
+            if message.content:
+                # If content is a list (e.g., list of dicts), extract text
+                if isinstance(message.content, list):
+                    texts = [item.get('text', '') for item in message.content if isinstance(item, dict) and 'text' in item]
+                    if texts:
+                        print(f"AI: {' '.join(texts)}")
+                elif isinstance(message.content, str):
+                    print(f"AI: {message.content}")
+        elif isinstance(message, SystemMessage):
+            print(f"System: {message.content}")
+        elif isinstance(message, ToolMessage):
+            print(f"Tool: {message.content}")
 
-    print(f"Tools called: {state.tool_call_required}")
+    print(f"Tools Enabled: {state.tools_enabled}")
     print(f"Search Enabled: {state.search_enabled}")
     print(f"Next Stage: {state.next_stage}")
 
+
     # Log boolean completion flags
     print(f"Idea Complete: {state.idea_complete}")
     print(f"Brainstorming Complete: {state.brainstorming_complete}")
@@ -244,7 +260,6 @@ async def brainstorming_node(state: GraphProcessingState, config=None):
     print(f"Drawing Complete: {state.drawing_complete}")
     print(f"Product Searching Complete: {state.product_searching_complete}")
     print(f"Purchasing Complete: {state.purchasing_complete}")
-    print("--- End Guidance Node Debug ---")
 
     # Check if model is available
     if not model:
@@ -452,10 +467,10 @@ def define_workflow() -> CompiledStateGraph:
     # workflow.add_edge("tools", "brainstorming_node")
 
     workflow.add_conditional_edges(
-        "brainstorming_node",
+        "guidance_node",
         tools_condition,
     )
-    workflow.add_conditional_edges("guidance_node", guidance_routing)
+    # workflow.add_conditional_edges("guidance_node", guidance_routing)
     # workflow.add_conditional_edges("brainstorming_node", brainstorming_routing)
 
     # # Set end nodes