Update app.py

app.py

@@ -3,7 +3,7 @@ import json
 import uuid
 import datetime
 import logging
-from flask import Flask, request, render_template_string, jsonify
+import re
 
 # Setup logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -18,9 +18,6 @@ except ImportError:
     AutoModelForCausalLM = None
     AutoTokenizer = None
 
-# Initialize Flask app
-app = Flask(__name__)
-
 # Initialize AI model (distilbert as placeholder; replace with fine-tuned model or Mistral-7B)
 model_name = "distilbert-base-uncased"  # Lightweight model for demo
 if TRANSFORMERS_AVAILABLE:
@@ -37,7 +34,7 @@ else:
     model = None
 
 # Database setup
-conn = sqlite3.connect("erp.db", check_same_thread=False)
+conn = sqlite3.connect("erp.db")
 cursor = conn.cursor()
 
 # Create tables
@@ -86,14 +83,20 @@ def initialize_chart_of_accounts():
         ("1.2", "Current Assets", "Asset", "1", True, False),
         ("1.2.1", "Cash", "Asset", "1.2", True, True),
         ("1.2.2", "Laptop", "Asset", "1.2", True, True),
+        ("1.2.3", "Inventory", "Asset", "1.2", True, True),
+        ("1.2.4", "Accounts Receivable", "Asset", "1.2", True, True),
         ("2", "Liabilities", "Liability", None, True, False),
         ("2.1", "Accounts Payable", "Liability", "2", True, True),
+        ("2.2", "Loan Payable", "Liability", "2", True, True),
         ("3", "Equity", "Equity", None, True, False),
-        ("3.1", "Owner's Capital", "Equity", "3", True, True),
+        ("3.1", "Owner's Equity", "Equity", "3", True, True),
+        ("3.2", "Drawings", "Equity", "3", True, True),
         ("4", "Revenue", "Revenue", None, True, False),
-        ("4.1", "Sales", "Revenue", "4", True, True),
+        ("4.1", "Sales Revenue", "Revenue", "4", True, True),
         ("5", "Expenses", "Expense", None, True, False),
-        ("5.1", "Operating Expenses", "Expense", "5", True, True)
+        ("5.1", "Rent Expense", "Expense", "5", True, True),
+        ("5.2", "Salary Expense", "Expense", "5", True, True),
+        ("5.3", "Office Supplies", "Expense", "5", True, True)
     ]
     cursor.executemany("""
     INSERT OR REPLACE INTO chart_of_accounts
@@ -103,7 +106,7 @@ def initialize_chart_of_accounts():
     conn.commit()
     logging.info("Chart of accounts initialized.")
 
-# Parse prompt using AI model (or fallback)
+# Enhanced fallback parser
 def parse_prompt(prompt):
     if model and tokenizer:
         try:
@@ -111,7 +114,7 @@ def parse_prompt(prompt):
             Parse the following accounting prompt into a JSON object with:
             - debit: {{account, type, amount}}
             - credit: {{account, type, amount}}
-            - payment_method: 'cash' or 'credit' or null
+            - payment_method: 'cash', 'credit', or null
             Prompt: {prompt}
             """
             inputs = tokenizer(input_text, return_tensors="pt")
@@ -121,34 +124,77 @@ def parse_prompt(prompt):
         except Exception as e:
             logging.warning(f"Model parsing failed: {e}. Using fallback parser.")
 
-    # Fallback parsing for common scenarios
-    prompt_lower = prompt.lower()
+    # Fallback parser
+    prompt_lower = prompt.lower().strip()
     amount = None
-    for word in prompt_lower.split():
-        if word.startswith("$"):
-            try:
-                amount = float(word[1:])
-                break
-            except:
-                pass
-    
+    # Extract amount
+    match = re.search(r'\$[\d,.]+', prompt_lower)
+    if match:
+        try:
+            amount = float(match.group().replace('$', '').replace(',', ''))
+        except ValueError:
+            logging.error("Invalid amount format.")
+            return None
+
     if not amount:
         logging.error("No amount found in prompt.")
         return None
 
-    if "laptop" in prompt_lower:
-        debit_account = "Laptop"
-        debit_type = "Asset"
-        if "cash" in prompt_lower:
-            credit_account = "Cash"
-            credit_type = "Asset"
-            payment_method = "cash"
-        elif "credit" in prompt_lower:
-            credit_account = "Accounts Payable"
-            credit_type = "Liability"
-            payment_method = "credit"
-        else:
-            return {"debit": {"account": "Laptop", "type": "Asset", "amount": amount}, "credit": None, "payment_method": None}
+    # Map keywords to accounts and types
+    account_mappings = {
+        "laptop": ("Laptop", "Asset"),
+        "inventory": ("Inventory", "Asset"),
+        "machinery": ("Machinery", "Asset"),
+        "building": ("Building", "Asset"),
+        "plant": ("Plant", "Asset"),
+        "office supplies": ("Office Supplies", "Expense"),
+        "cash": ("Cash", "Asset"),
+        "receivable": ("Accounts Receivable", "Asset"),
+        "sold goods": ("Sales Revenue", "Revenue"),
+        "sales": ("Sales Revenue", "Revenue"),
+        "rent": ("Rent Expense", "Expense"),
+        "salary": ("Salary Expense", "Expense"),
+        "paid": ("Cash", "Asset"),
+        "bought": ("Laptop", "Asset"),  # Default to Laptop; refine below
+        "purchased": ("Laptop", "Asset")
+    }
+
+    debit_account = None
+    debit_type = None
+    credit_account = None
+    credit_type = None
+    payment_method = None
+
+    # Determine debit account
+    for keyword, (account, acc_type) in account_mappings.items():
+        if keyword in prompt_lower:
+            if keyword in ["bought", "purchased"]:
+                # Check for specific asset/expense
+                for asset in ["laptop", "inventory", "machinery", "building", "plant", "office supplies"]:
+                    if asset in prompt_lower:
+                        debit_account, debit_type = account_mappings[asset]
+                        break
+                if not debit_account:
+                    debit_account, debit_type = account, acc_type
+            elif keyword in ["rent", "salary", "office supplies"]:
+                debit_account, debit_type = account, acc_type
+            elif keyword in ["sold goods", "sales"]:
+                debit_account, debit_type = "Accounts Receivable", "Asset"
+                credit_account, credit_type = account, acc_type
+            break
+
+    # Determine credit account and payment method
+    if "cash" in prompt_lower:
+        credit_account, credit_type = "Cash", "Asset"
+        payment_method = "cash"
+    elif "credit" in prompt_lower:
+        credit_account, credit_type = "Accounts Payable", "Liability"
+        payment_method = "credit"
+    elif debit_account and not credit_account:
+        # Ambiguous payment method
+        return {"debit": {"account": debit_account, "type": debit_type, "amount": amount}, "credit": None, "payment_method": None}
+
+    if debit_account and credit_account:
         return {
             "debit": {"account": debit_account, "type": debit_type, "amount": amount},
             "credit": {"account": credit_account, "type": credit_type, "amount": amount},
@@ -169,20 +215,23 @@ def generate_journal_entry(prompt, follow_up_response=None):
 
     # Handle ambiguous payment method
     if not payment_method and not follow_up_response:
-        return {"status": "clarify", "message": "Wonderful, did you buy on credit? (Yes/No)", "original_prompt": prompt}
+        return {"status": "clarify", "message": "Was this bought on cash or credit? (cash/credit)", "original_prompt": prompt}
 
     # Determine credit account
     credit_account = None
     credit_type = None
-    if follow_up_response and follow_up_response.lower() == "yes":
-        credit_account = "Accounts Payable"
-        credit_type = "Liability"
+    if follow_up_response:
+        follow_up_lower = follow_up_response.lower()
+        if follow_up_lower == "cash":
+            credit_account, credit_type = "Cash", "Asset"
+        elif follow_up_lower == "credit":
+            credit_account, credit_type = "Accounts Payable", "Liability"
+        else:
+            return "Invalid response. Please specify 'cash' or 'credit'."
     elif payment_method == "cash":
-        credit_account = parsed["credit"]["account"]
-        credit_type = parsed["credit"]["type"]
+        credit_account, credit_type = parsed["credit"]["account"], parsed["credit"]["type"]
     elif payment_method == "credit":
-        credit_account = "Accounts Payable"
-        credit_type = "Liability"
+        credit_account, credit_type = "Accounts Payable", "Liability"
     else:
         return "Invalid payment method specified."
 
@@ -249,91 +298,46 @@ def generate_t_account(account_name):
     
     return t_account
 
-# HTML template for web interface
-HTML_TEMPLATE = """
-<!DOCTYPE html>
-<html>
-<head>
-    <title>AI ERP System</title>
-    <style>
-        body { font-family: Arial, sans-serif; margin: 20px; }
-        h1 { color: #333; }
-        textarea, input { width: 100%; margin: 10px 0; }
-        pre { background: #f4f4f4; padding: 10px; white-space: pre-wrap; }
-        .error { color: red; }
-    </style>
-</head>
-<body>
-    <h1>AI ERP System</h1>
-    <h2>Enter Transaction Prompt</h2>
-    <form method="POST" action="/process_prompt">
-        <textarea name="prompt" rows="4" placeholder="e.g., Bought a laptop for $200 on cash"></textarea>
-        <input type="submit" value="Submit Prompt">
-    </form>
-    {% if result %}
-        <h2>Result</h2>
-        {% if result.status == 'clarify' %}
-            <p>{{ result.message }}</p>
-            <form method="POST" action="/process_follow_up">
-                <input type="hidden" name="original_prompt" value="{{ result.original_prompt }}">
-                <input type="text" name="follow_up" placeholder="Yes/No">
-                <input type="submit" value="Submit Response">
-            </form>
-        {% else %}
-            <pre>{{ result }}</pre>
-        {% endif %}
-    {% endif %}
-    <h2>View T-Account</h2>
-    <form method="POST" action="/t_account">
-        <input type="text" name="account_name" placeholder="e.g., Laptop">
-        <input type="submit" value="Generate T-Account">
-    </form>
-    {% if t_account %}
-        <h2>T-Account</h2>
-        <pre>{{ t_account }}</pre>
-    {% endif %}
-</body>
-</html>
-"""
-
-# Flask routes
-@app.route("/", methods=["GET", "POST"])
-def index():
+# Main CLI loop
+def main():
     initialize_chart_of_accounts()
-    return render_template_string(HTML_TEMPLATE, result=None, t_account=None)
-
-@app.route("/process_prompt", methods=["POST"])
-def process_prompt():
-    prompt = request.form.get("prompt")
-    if not prompt:
-        return render_template_string(HTML_TEMPLATE, result="No prompt provided.", t_account=None)
+    print("AI ERP System: Enter accounting prompts or 't-account <account_name>' to view T-accounts. Type 'exit' to quit.")
     
-    result = generate_journal_entry(prompt)
-    return render_template_string(HTML_TEMPLATE, result=result, t_account=None)
-
-@app.route("/process_follow_up", methods=["POST"])
-def process_follow_up():
-    original_prompt = request.form.get("original_prompt")
-    follow_up = request.form.get("follow_up")
-    if not original_prompt or not follow_up:
-        return render_template_string(HTML_TEMPLATE, result="Missing prompt or response.", t_account=None)
-    
-    result = generate_journal_entry(original_prompt, follow_up)
-    return render_template_string(HTML_TEMPLATE, result=result, t_account=None)
-
-@app.route("/t_account", methods=["POST"])
-def t_account():
-    account_name = request.form.get("account_name")
-    if not account_name:
-        return render_template_string(HTML_TEMPLATE, result=None, t_account="No account name provided.")
+    pending_prompt = None
+    while True:
+        try:
+            prompt = input("> ").strip()
+            if prompt.lower() == "exit":
+                break
+            if prompt.lower().startswith("t-account "):
+                account_name = prompt[10:].strip()
+                if account_name:
+                    print(generate_t_account(account_name))
+                else:
+                    print("Please specify an account name.")
+                continue
+            
+            # Handle follow-up response
+            if pending_prompt:
+                result = generate_journal_entry(pending_prompt, prompt)
+                pending_prompt = None
+            else:
+                result = generate_journal_entry(prompt)
+            
+            if isinstance(result, dict) and result["status"] == "clarify":
+                print(result["message"])
+                pending_prompt = result["original_prompt"]
+            else:
+                print(result)
+        
+        except KeyboardInterrupt:
+            print("\nExiting...")
+            break
+        except Exception as e:
+            logging.error(f"Error processing prompt: {e}")
+            print("An error occurred. Please try again.")
     
-    t_account = generate_t_account(account_name)
-    return render_template_string(HTML_TEMPLATE, result=None, t_account=t_account)
+    conn.close()
 
-# Run the app
 if __name__ == "__main__":
-    # Initialize chart of accounts
-    initialize_chart_of_accounts()
-    
-    # Run Flask app (use port 7860 for Hugging Face Spaces)
-    app.run(host="0.0.0.0", port=7860, debug=False)
+    main()