Update app.py

app.py

@@ -306,11 +306,12 @@ def chat_function(message, history, state=None):
     if message.lower().startswith("t-account "):
         account_name = message[10:].strip()
         if account_name:
-            return {"role": "assistant", "content": generate_t_account(account_name)}, state
-        return {"role": "assistant", "content": "Please specify an account name."}, state
-    
-    parsed, state = parse_prompt(message, state)
-    response, state = generate_journal_entry(parsed, state)
+            response = generate_t_account(account_name)
+        else:
+            response = "Please specify an account name."
+    else:
+        parsed, state = parse_prompt(message, state)
+        response, state = generate_journal_entry(parsed, state)
     
     if history is not None:
         history.append({"role": "user", "content": message})
@@ -321,20 +322,20 @@ def chat_function(message, history, state=None):
             {"role": "assistant", "content": response}
         ]
     
-    return history, state
+    return history, state, ""  # Clear the textbox by returning an empty string
 
 # Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# AI ERP System")
     gr.Markdown("Enter accounting prompts like 'Bought a laptop for $200' or 't-account Laptop'. The system will ask for clarification if needed.")
     chatbot = gr.Chatbot(type="messages")
-    msg = gr.Textbox(placeholder="Type your prompt here...", lines=2)
+    msg = gr.Textbox(placeholder="Type your prompt here...", lines=1, submit_btn=None)
     clear = gr.Button("Clear")
     
     state = gr.State({})
     
-    msg.submit(chat_function, [msg, chatbot, state], [chatbot, state])
-    clear.click(lambda: ([], {}), None, [chatbot, state], queue=False)
+    msg.submit(chat_function, [msg, chatbot, state], [chatbot, state, msg])
+    clear.click(lambda: ([], {}, ""), None, [chatbot, state, msg], queue=False)
 
 # Launch Gradio
 if __name__ == "__main__":