Update app.py

app.py

@@ -109,7 +109,6 @@ def parse_prompt(prompt, state):
         except Exception as e:
             logging.error(f"Model parsing failed: {e}")
 
-    # Fallback parser
     prompt_lower = prompt.lower().strip()
     amount = None
     match = re.search(r'\$[\d,.]+', prompt_lower)
@@ -122,7 +121,6 @@ def parse_prompt(prompt, state):
     if not amount:
         return {"error": "No amount found in prompt."}, state
 
-    # Account mappings
     account_mappings = {
         "laptop": ("Laptop", "Asset"),
         "inventory": ("Inventory", "Asset"),
@@ -150,7 +148,6 @@ def parse_prompt(prompt, state):
     credit_type = None
     payment_method = None
 
-    # Handle follow-up response
     if state.get("pending_prompt"):
         follow_up = prompt_lower
         if follow_up in ["cash", "credit", "bank"]:
@@ -168,9 +165,7 @@ def parse_prompt(prompt, state):
             state = {}  # Clear state
         else:
             return {"error": "Please respond with 'cash', 'credit', or 'bank'."}, state
-
     else:
-        # Parse new prompt
         for keyword, (account, acc_type) in account_mappings.items():
             if keyword in prompt_lower:
                 if keyword in ["bought", "purchased"]:
@@ -235,7 +230,6 @@ def generate_journal_entry(parsed, state):
     credit_account = parsed["credit"]["account"]
     credit_type = parsed["credit"]["type"]
 
-    # Validate accounts
     cursor.execute("SELECT account_id, account_type, allow_posting FROM chart_of_accounts WHERE account_name = ?", (debit_account,))
     debit_result = cursor.fetchone()
     cursor.execute("SELECT account_id, account_type, allow_posting FROM chart_of_accounts WHERE account_name = ?", (credit_account,))
@@ -248,7 +242,6 @@ def generate_journal_entry(parsed, state):
     if debit_result[1] != parsed["debit"]["type"] or credit_result[1] != credit_type:
         return "Account type mismatch.", state
 
-    # Create journal entry
     entry_id = str(uuid.uuid4())
     date = datetime.datetime.now().isoformat()
     description = state.get("pending_prompt", "Transaction")
@@ -310,19 +303,25 @@ def chat_function(message, history, state=None):
     
     logging.info(f"Received message: {message}")
     
-    # Handle T-account request
     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
     
-    # Parse prompt and generate entry
     parsed, state = parse_prompt(message, state)
     response, state = generate_journal_entry(parsed, state)
     
-    # Format response for Gradio
-    return {"role": "assistant", "content": response}, state
+    if history is not None:
+        history.append({"role": "user", "content": message})
+        history.append({"role": "assistant", "content": response})
+    else:
+        history = [
+            {"role": "user", "content": message},
+            {"role": "assistant", "content": response}
+        ]
+    
+    return history, state
 
 # Gradio interface
 with gr.Blocks() as demo:
@@ -332,11 +331,10 @@ with gr.Blocks() as demo:
     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])
-    clear.click(lambda: ({"role": "assistant", "content": None}, []), None, [chatbot, state], queue=False)
+    msg.submit(chat_function, [msg, chatbot, state], [chatbot, state], _js="() => {}")
+    clear.click(lambda: ([], {}), None, [chatbot, state], queue=False)
 
 # Launch Gradio
 if __name__ == "__main__":