Spaces:

ZeeAI1
/

abctest

Sleeping

App Files Files Community

ZeeAI1 commited on May 8

Commit

ab73613

verified ·

1 Parent(s): 2de96a2

Update app.py

Browse files

Files changed (1) hide show

app.py +88 -19

app.py CHANGED Viewed

@@ -3,6 +3,7 @@ import json
 import uuid
 import datetime
 import logging
 # Setup logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -17,6 +18,9 @@ except ImportError:
     AutoModelForCausalLM = None
     AutoTokenizer = None
 # 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:
@@ -33,7 +37,7 @@ else:
     model = None
 # Database setup
-conn = sqlite3.connect("erp.db")
 cursor = conn.cursor()
 # Create tables
@@ -165,7 +169,7 @@ 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)"}
     # Determine credit account
     credit_account = None
@@ -245,26 +249,91 @@ def generate_t_account(account_name):
     return t_account
-# Example usage
-if __name__ == "__main__":
     initialize_chart_of_accounts()
-    # Test prompt (cash purchase)
-    prompt = "Bought a laptop for $200 on cash"
     result = generate_journal_entry(prompt)
-    print(result)
-    # Test prompt (ambiguous payment method)
-    prompt = "Bought a laptop for $300"
-    result = generate_journal_entry(prompt)
-    print(result)
-    if isinstance(result, dict) and result["status"] == "clarify":
-        result = generate_journal_entry(prompt, "Yes")
-        print(result)
-    # Test T-account
-    t_account = generate_t_account("Laptop")
-    print(t_account)
-    # Clean up
-    conn.close()

 import uuid
 import datetime
 import logging
+from flask import Flask, request, render_template_string, jsonify
 # Setup logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
     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:
     model = None
 # Database setup
+conn = sqlite3.connect("erp.db", check_same_thread=False)
 cursor = conn.cursor()
 # Create tables
     # 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}
     # Determine credit account
     credit_account = None
     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():
     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)
     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.")
+    t_account = generate_t_account(account_name)
+    return render_template_string(HTML_TEMPLATE, result=None, t_account=t_account)
+# 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)