157 lines
4.1 KiB
JavaScript
157 lines
4.1 KiB
JavaScript
|
|
/*const express = require("express");
|
|
const cors = require("cors");
|
|
const dotenv = require("dotenv");
|
|
const connectDB = require("../config/db");
|
|
const http = require("http");
|
|
const TransRoutes = require("./routes/")
|
|
dotenv.config();
|
|
const app = express();
|
|
const server = http.createServer(app); // Pass `app` to `http.createServer`
|
|
|
|
// Connect Database
|
|
connectDB();
|
|
// Middleware
|
|
app.use(express.json());
|
|
app.use(cors());
|
|
|
|
// Routes
|
|
app.use("/api/transactions", TransRoutes);
|
|
|
|
// Start Server
|
|
const PORT = process.env.PORT || 5000;
|
|
server.listen(PORT, () => {
|
|
console.log(`✅ Server running on http://localhost:${PORT}`);
|
|
});
|
|
*/
|
|
/*
|
|
const express = require('express');
|
|
const mongoose = require('mongoose');
|
|
const app = express();
|
|
app.use(express.json()); // to parse JSON
|
|
const predictRoute = require('./routes/predict');
|
|
app.use('/api/predict', predictRoute);
|
|
mongoose.connect('mongodb://localhost:27017/chat', { useNewUrlParser: true, useUnifiedTopology: true });
|
|
|
|
// Define schema and model
|
|
const transactionSchema = new mongoose.Schema({
|
|
description: String,
|
|
amount: Number,
|
|
date: { type: Date, default: Date.now },
|
|
});
|
|
|
|
const Transaction = mongoose.model('Transaction', transactionSchema);
|
|
|
|
// POST endpoint to add transaction
|
|
app.post('/api/transaction/add', async (req, res) => {
|
|
try {
|
|
const newTransaction = new Transaction(req.body);
|
|
await newTransaction.save();
|
|
res.status(200).json({ message: 'Transaction saved', data: newTransaction });
|
|
} catch (err) {
|
|
res.status(500).json({ error: err.message });
|
|
}
|
|
});
|
|
|
|
app.listen(3000, () => {
|
|
console.log('Server running on port 3000');
|
|
});
|
|
*/
|
|
const express = require('express');
|
|
const mongoose = require('mongoose');
|
|
const axios = require('axios');
|
|
|
|
const app = express();
|
|
app.use(express.json()); // to parse JSON
|
|
|
|
// Prediction route
|
|
const predictRoute = require('./routes/predict');
|
|
app.use('/api/predict', predictRoute);
|
|
|
|
// MongoDB connection
|
|
mongoose.connect('mongodb://localhost:27017/chat', {
|
|
useNewUrlParser: true,
|
|
useUnifiedTopology: true,
|
|
});
|
|
|
|
// Define schema and model
|
|
const transactionSchema = new mongoose.Schema({
|
|
description: String,
|
|
amount: Number,
|
|
|
|
category: String, // 👈 added category
|
|
date: { type: Date, default: Date.now },
|
|
});
|
|
|
|
const Transaction = mongoose.model('Transaction', transactionSchema);
|
|
|
|
// POST endpoint to add transaction with auto-categorization
|
|
app.post('/api/transaction/add', async (req, res) => {
|
|
try {
|
|
const { description, amount } = req.body;
|
|
|
|
// Call your AI prediction endpoint
|
|
const predictRes = await axios.post('http://192.168.1.9:5000/api/predict', {
|
|
description,
|
|
});
|
|
|
|
const category = predictRes.data.category || 'Other';
|
|
|
|
const newTransaction = new Transaction({
|
|
description,
|
|
amount,
|
|
category,
|
|
});
|
|
|
|
await newTransaction.save();
|
|
|
|
res.status(200).json({
|
|
message: '✅ Transaction saved with category',
|
|
data: newTransaction,
|
|
});
|
|
} catch (err) {
|
|
console.error('❌ Error adding transaction:', err.message);
|
|
res.status(500).json({ error: err.message });
|
|
}
|
|
});
|
|
app.post('/api/transaction/update', async (req, res) => {
|
|
try {
|
|
const { budget } = req.body;
|
|
// Save to database or wherever you store budgets
|
|
await BudgetModel.updateOne({}, { amount: budget }, { upsert: true });
|
|
res.status(200).json({ success: true });
|
|
} catch (error) {
|
|
res.status(500).json({ error: 'Failed to update budget' });
|
|
}
|
|
});
|
|
// GET transactions by category
|
|
app.get('/api/transactions', async (req, res) => {
|
|
try {
|
|
const category = req.query.category;
|
|
|
|
let query = {};
|
|
if (category && category !== 'All') {
|
|
query.category = category;
|
|
}
|
|
|
|
const transactions = await Transaction.find(query).sort({ date: -1 });
|
|
|
|
res.status(200).json({
|
|
success: true,
|
|
data: transactions.map(tx => ({
|
|
id: tx._id,
|
|
description: tx.description,
|
|
amount: tx.amount,
|
|
category: tx.category,
|
|
date: tx.date,
|
|
})),
|
|
});
|
|
} catch (err) {
|
|
res.status(500).json({ error: err.message });
|
|
}
|
|
});
|
|
|
|
// Start server
|
|
app.listen(3000, () => {
|
|
console.log('🚀 Server running on port 3000');
|
|
});
|