Update app.py

app.py

@@ -86,8 +86,6 @@ async def chat_fn(user_input: str, history: list, input_graph_state: dict, uuid:
     try:
         logger.info(f"Processing user input: {user_input[:100]}...")
         logger.info(f"History format: {type(history)}, length: {len(history) if history else 0}")
-        if history:
-            logger.info(f"Sample history entry: {history[0] if len(history) > 0 else 'None'}")
         
         # Initialize input_graph_state if None
         if input_graph_state is None:
@@ -132,7 +130,7 @@ async def chat_fn(user_input: str, history: list, input_graph_state: dict, uuid:
         )
 
         output: str = ""
-        final_state: dict = {}
+        final_state: dict = input_graph_state.copy()  # Initialize with current state
         waiting_output_seq: list[str] = []
         
         # Add user message to history immediately
@@ -158,14 +156,14 @@ async def chat_fn(user_input: str, history: list, input_graph_state: dict, uuid:
                                 # Update the last tuple with current status
                                 if updated_history:
                                     updated_history[-1] = (user_input, "\n".join(waiting_output_seq))
-                                yield updated_history, gr.skip(), gr.skip()
+                                yield updated_history, final_state, False
 
                             elif tool_name == "download_website_text":
                                 url = msg_tool_call['args']['url']
                                 waiting_output_seq.append(f"📥 Downloading text from '{url}'...")
                                 if updated_history:
                                     updated_history[-1] = (user_input, "\n".join(waiting_output_seq))
-                                yield updated_history, gr.skip(), gr.skip()
+                                yield updated_history, final_state, False
 
                             elif tool_name == "human_assistance":
                                 query = msg_tool_call["args"]["query"]
@@ -185,7 +183,7 @@ async def chat_fn(user_input: str, history: list, input_graph_state: dict, uuid:
                                 waiting_output_seq.append(f"🔧 Running {tool_name}...")
                                 if updated_history:
                                     updated_history[-1] = (user_input, "\n".join(waiting_output_seq))
-                                yield updated_history, gr.skip(), gr.skip()
+                                yield updated_history, final_state, False
 
             elif stream_mode == "messages":
                 msg, metadata = chunk
@@ -207,13 +205,13 @@ async def chat_fn(user_input: str, history: list, input_graph_state: dict, uuid:
                         # Update the last tuple with accumulated output
                         if updated_history:
                             updated_history[-1] = (user_input, output)
-                        yield updated_history, gr.skip(), gr.skip()
+                        yield updated_history, final_state, False
 
         # Final yield with complete response
         if updated_history:
-            updated_history[-1] = (user_input, output + " ")
+            updated_history[-1] = (user_input, output.strip() if output else "I'm here to help with your DIY projects!")
         logger.info(f"Final response: {output[:100]}...")
-        yield updated_history, dict(final_state), True
+        yield updated_history, final_state, True
         
     except Exception as e:
         logger.exception("Exception occurred in chat_fn")
@@ -221,7 +219,8 @@ async def chat_fn(user_input: str, history: list, input_graph_state: dict, uuid:
         if not isinstance(history, list):
             history = []
         error_history = history + [(user_input, error_message)]
-        yield error_history, gr.skip(), False
+        # Return safe values instead of gr.skip()
+        yield error_history, input_graph_state or {}, False
 
 
 def convert_to_tuples_format(messages_list):