Create app.py
Browse files
app.py
ADDED
|
@@ -0,0 +1,73 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import json
|
| 2 |
+
import xmlrpc.client
|
| 3 |
+
from transformers import pipeline
|
| 4 |
+
|
| 5 |
+
# --- CONFIGURATION ---
|
| 6 |
+
ODOO_URL = "https://your-odoo-instance.com"
|
| 7 |
+
ODOO_DB = "your-database"
|
| 8 |
+
ODOO_USERNAME = "your-username"
|
| 9 |
+
ODOO_PASSWORD = "your-password"
|
| 10 |
+
ACCOUNT_ID = 41 # Replace with correct expense account ID
|
| 11 |
+
|
| 12 |
+
# --- 1. LOAD NLP MODEL ---
|
| 13 |
+
print("Loading Hugging Face model...")
|
| 14 |
+
model = pipeline("text2text-generation", model="google/flan-t5-base")
|
| 15 |
+
|
| 16 |
+
# --- 2. GET USER PROMPT ---
|
| 17 |
+
user_prompt = input("Enter your accounting prompt:\n> ")
|
| 18 |
+
|
| 19 |
+
# --- 3. NLP PROCESSING ---
|
| 20 |
+
formatted_prompt = (
|
| 21 |
+
f"Extract as JSON with fields transaction_type, vendor_name, amount, date, description from: {user_prompt}"
|
| 22 |
+
)
|
| 23 |
+
|
| 24 |
+
result = model(formatted_prompt, max_length=256)[0]['generated_text']
|
| 25 |
+
print("\nRaw Model Output:\n", result)
|
| 26 |
+
|
| 27 |
+
# --- 4. PARSE JSON ---
|
| 28 |
+
try:
|
| 29 |
+
output_data = json.loads(result)
|
| 30 |
+
except json.JSONDecodeError:
|
| 31 |
+
print("\nERROR: Failed to parse output as JSON. Please adjust model or prompt.")
|
| 32 |
+
exit(1)
|
| 33 |
+
|
| 34 |
+
# Debug print
|
| 35 |
+
print("\nParsed Data:\n", output_data)
|
| 36 |
+
|
| 37 |
+
vendor_name = output_data['vendor_name']
|
| 38 |
+
amount = float(output_data['amount'])
|
| 39 |
+
invoice_date = output_data['date']
|
| 40 |
+
description = output_data['description']
|
| 41 |
+
|
| 42 |
+
# --- 5. CONNECT TO ODOO ---
|
| 43 |
+
print("\nConnecting to Odoo...")
|
| 44 |
+
common = xmlrpc.client.ServerProxy(f"{ODOO_URL}/xmlrpc/2/common")
|
| 45 |
+
uid = common.authenticate(ODOO_DB, ODOO_USERNAME, ODOO_PASSWORD, {})
|
| 46 |
+
models = xmlrpc.client.ServerProxy(f"{ODOO_URL}/xmlrpc/2/object")
|
| 47 |
+
|
| 48 |
+
# --- 6. FIND OR CREATE VENDOR ---
|
| 49 |
+
partner_id = models.execute_kw(ODOO_DB, uid, ODOO_PASSWORD, 'res.partner', 'search', [[['name', '=', vendor_name]]])
|
| 50 |
+
if not partner_id:
|
| 51 |
+
partner_id = models.execute_kw(ODOO_DB, uid, ODOO_PASSWORD, 'res.partner', 'create', [{
|
| 52 |
+
'name': vendor_name,
|
| 53 |
+
'supplier_rank': 1
|
| 54 |
+
}])
|
| 55 |
+
print(f"Vendor '{vendor_name}' created with ID {partner_id}")
|
| 56 |
+
else:
|
| 57 |
+
partner_id = partner_id[0]
|
| 58 |
+
print(f"Vendor '{vendor_name}' found with ID {partner_id}")
|
| 59 |
+
|
| 60 |
+
# --- 7. CREATE VENDOR BILL ---
|
| 61 |
+
invoice_id = models.execute_kw(ODOO_DB, uid, ODOO_PASSWORD, 'account.move', 'create', [{
|
| 62 |
+
'move_type': 'in_invoice',
|
| 63 |
+
'partner_id': partner_id,
|
| 64 |
+
'invoice_date': invoice_date,
|
| 65 |
+
'invoice_line_ids': [(0, 0, {
|
| 66 |
+
'name': description,
|
| 67 |
+
'quantity': 1,
|
| 68 |
+
'price_unit': amount,
|
| 69 |
+
'account_id': ACCOUNT_ID,
|
| 70 |
+
})],
|
| 71 |
+
}])
|
| 72 |
+
|
| 73 |
+
print(f"\nVendor Bill Created Successfully in Odoo! Invoice ID: {invoice_id}")
|