Added new features to improve performance.

app.py

@@ -1,18 +1,30 @@
 import gradio as gr
 from indexer import run_email_agent
 
-with gr.Blocks() as demo:
+with gr.Blocks(css="""
+    body {background-color: #fffaf0;}  /* Light creme background */
+    .gr-textbox {min-height: 200px;}  /* Make textboxes taller */
+""") as demo:
     gr.Markdown("## Tool-Based AI Email Agent")
 
-    user_input = gr.Textbox(label="User Query")
+    user_input = gr.Textbox(
+        label="User Query",
+        placeholder="Type your question here...",
+        lines=4  # make input box slightly bigger
+    )
     submit_btn = gr.Button("Run Agent")
 
-    output = gr.Textbox(label="Agent Response")
+    output = gr.Textbox(
+        label="Agent Response",
+        placeholder="The agent will respond here...",
+        lines=10  # bigger response box
+    )
 
     human_box = gr.Textbox(
         label="Human Assistance Required",
         visible=False
     )
+
     def handle_query(text):
         response = run_email_agent(text)
         return response
@@ -23,4 +35,4 @@ with gr.Blocks() as demo:
         outputs=output
     )
 
-demo.launch()
+demo.launch()

indexer.py

@@ -100,12 +100,16 @@ graph_builder.add_node("tools", tool_node)
 graph_builder.add_edge(START, "chatbot")
 
 graph_builder.add_conditional_edges(
-    "chatbot", tools_condition
+    "chatbot",
+    tools_condition,
+    {
+        "tools": "tools",
+        "__end__": END
+    }
 )
 
-graph_builder.add_edge("tools","chatbot")
 
-graph_builder.add_edge("chatbot",END)
+graph_builder.add_edge("tools","chatbot")
 
 graph=graph_builder.compile()