clean code

app.py

@@ -263,7 +263,7 @@ No need to say if the chart cannot be displayed, because it already handled in t
 You already have access to a DataFrame called `df`
 """)
 
-
+#--- Assistant Functions ---
 def assistant(state: AgentState) -> AgentState:
 
     schema_output = describe_schema.invoke(df) 
@@ -438,38 +438,7 @@ def build_graph():
 react_graph = build_graph()
 config = {"configurable": {"thread_id": 123, "session": 100}}
 
-# --- Gradio UI ---
-def respond(message, chat_history):
-    chat_history = []
-    res = react_graph.invoke(
-        {"messages": [HumanMessage(content=message)]}
-        , config=config)
-
-    for msg in res["messages"]:
-        msg.pretty_print()
-        if isinstance(msg, HumanMessage):
-            chat_history.append({"role": "user", "content": msg.content})
-
-        if isinstance(msg, AIMessage):
-            ai_response = msg.content
-            chat_history.append({"role": "assistant", "content": ai_response})
-
-        if isinstance(msg, ToolMessage):
-            if msg.name == "generate_plot_code":
-                plot_result = generate_plot_from_code(msg.content)
-                chat_history.append({"role": "assistant", "content": gr.Plot(plot_result)})
-                chat_history.append({"role": "assistant", "content": res["plots"][-1].get("interpretation", " ")})
-
-            if msg.name == "enhance_plot_code":
-                plot_result = generate_plot_from_code(msg.content)
-                chat_history.append({"role": "assistant", "content": gr.Plot(plot_result)})
-
-    time.sleep(1)
-    return "", chat_history
-
-
-my_theme = gr.Theme.from_hub("NoCrypt/miku")
-
+# --- Data Exploration Functions ---
 def to_snake_case(name):
     return name.lower().replace(' ', '_').replace('-', '_')
 
@@ -576,7 +545,6 @@ def refresh_graph():
     global config
     config = {"configurable": {"thread_id": str(uuid.uuid4()), "session": str(uuid.uuid4())}}
 
-# --- Gradio UI ---
 def respond(message, chat_history):
     chat_history = []
     res = react_graph.invoke(
@@ -604,7 +572,8 @@ def respond(message, chat_history):
     time.sleep(1)
     return "", chat_history
 
-# Layout
+# --- Gradio UI ---
+my_theme = gr.Theme.from_hub("NoCrypt/miku")
 with gr.Blocks(theme=my_theme) as demo:
     demo.load(refresh_graph, inputs=None, outputs=None)