| import os |
| import gradio as gr |
| import sqlite3 |
| from dotenv import load_dotenv |
| import openai |
| from utils import generate_double_entry, save_transaction |
|
|
| |
| load_dotenv() |
| openai.api_key = os.getenv(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) |
| openai.api_base = "https://api.groq.com/openai/v1" |
|
|
| def call_groq(prompt): |
| response = openai.ChatCompletion.create( |
| model="mixtral-8x7b-instruct", |
| messages=[{"role": "user", "content": prompt}] |
| ) |
| return response['choices'][0]['message']['content'] |
|
|
| def accounting_chatbot(user_input): |
| llm_output = call_groq(user_input) |
| debit, credit, amount = generate_double_entry(llm_output) |
| save_transaction(debit, credit, amount, user_input) |
| |
| output = f"**Double Entry**\\n- Debit: {debit}\\n- Credit: {credit}\\n- Amount: {amount}\\n\\n🧠 {llm_output}" |
| return output |
|
|
| demo = gr.ChatInterface(fn=accounting_chatbot, title="AI Accounting Assistant") |
| demo.launch() |
|
|
