File size: 2,323 Bytes
9042ecd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
# app.py
import streamlit as st
from transformers import pipeline
from datetime import datetime
import uuid
# Initialize Hugging Face pipeline
prompt_generator = pipeline("text-generation", model="gpt2")
# Basic structure for storing transactions
if 'transactions' not in st.session_state:
st.session_state.transactions = []
# App Title
st.title("AI Prompt-based ERP System")
# User onboarding and initial configuration
st.sidebar.header("Business Setup")
company_name = st.sidebar.text_input("Company Name")
industry = st.sidebar.text_input("Industry")
if st.sidebar.button("Generate Chart of Accounts"):
coa_prompt = f"Generate a basic chart of accounts for a {industry} company named {company_name}."
coa = prompt_generator(coa_prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
st.sidebar.write("**Suggested Chart of Accounts:**")
st.sidebar.write(coa)
# Transaction input
st.header("Record a Transaction")
user_prompt = st.text_area("Describe your transaction (e.g., 'I paid $30,000 in shop tax today'):")
if st.button("Generate Accounting Entry") and user_prompt:
full_prompt = f"Create accounting entry from prompt: '{user_prompt}'"
generated_entry = prompt_generator(full_prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
# Generate unique transaction ID
transaction_id = str(uuid.uuid4())[:8]
# Record transaction
transaction = {
"id": transaction_id,
"date": datetime.now().strftime("%Y-%m-%d"),
"description": user_prompt,
"entry": generated_entry
}
st.session_state.transactions.append(transaction)
st.success(f"Transaction recorded with ID: {transaction_id}")
# Display recorded transactions
st.header("Transaction Records")
if st.session_state.transactions:
for trans in st.session_state.transactions:
st.subheader(f"Transaction ID: {trans['id']}")
st.write(f"**Date:** {trans['date']}")
st.write(f"**Description:** {trans['description']}")
st.write(f"**Generated Entry:** {trans['entry']}")
st.markdown("---")
else:
st.write("No transactions recorded yet.")
# Option to clear transactions
if st.button("Clear All Transactions"):
st.session_state.transactions.clear()
st.warning("All transactions cleared.")
|