Update app.py

app.py

@@ -314,35 +314,29 @@ def chat_function(message, history, state=None):
     if message.lower().startswith("t-account "):
         account_name = message[10:].strip()
         if account_name:
-            return generate_t_account(account_name), state
-        return "Please specify an account name.", state
+            return {"role": "assistant", "content": generate_t_account(account_name)}, state
+        return {"role": "assistant", "content": "Please specify an account name."}, state
     
     # Parse prompt and generate entry
     parsed, state = parse_prompt(message, state)
     response, state = generate_journal_entry(parsed, state)
     
-    # Append to history
-    if history is not None:
-        history.append((message, response))
-    else:
-        history = [(message, response)]
-    
-    logging.info(f"Response: {response}")
-    return response, state
+    # Format response for Gradio
+    return {"role": "assistant", "content": response}, state
 
 # 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()
+    chatbot = gr.Chatbot(type="messages")
     msg = gr.Textbox(placeholder="Type your prompt here...", lines=2)
     clear = gr.Button("Clear")
     
     # Maintain state
     state = gr.State({})
     
-    msg.submit(chat_function, [msg, chatbot, state], [chatbot, state], _js="() => {return false;}")
-    clear.click(lambda: (None, []), None, [chatbot, state], queue=False)
+    msg.submit(chat_function, [msg, chatbot, state], [chatbot, state])
+    clear.click(lambda: ({"role": "assistant", "content": None}, []), None, [chatbot, state], queue=False)
 
 # Launch Gradio
 if __name__ == "__main__":