🤹 AI IMG RECOGNITION БЛЯТЬ

This commit is contained in:
Wind-Explorer
2024-08-12 22:42:34 +08:00
parent 6b7b4bba11
commit 878e854be8
7 changed files with 527 additions and 239 deletions

View File

@@ -0,0 +1,128 @@
import React, { useState } from "react";
import config from "../config";
import instance from "../security/http";
import { AxiosResponse } from "axios";
import { Card } from "@nextui-org/react";
interface InsertImageProps {
label: string;
onImageSelected: (file: File | null) => void;
onAmountResolved: (amount: number) => void;
onAiProcessingChange: React.Dispatch<React.SetStateAction<boolean>>;
}
const InsertBillImage: React.FC<InsertImageProps> = ({
label,
onImageSelected,
onAmountResolved,
onAiProcessingChange,
}) => {
const [selectedFile, setSelectedFile] = useState<File | null>(null);
const [previewImage, setPreviewImage] = useState<string>("");
const base64StringToChunks = (base64String: string): string[] => {
const chunks: string[] = [];
const chunkSize = 4096;
let offset = 0;
while (offset < base64String.length) {
const chunk = base64String.slice(offset, offset + chunkSize);
chunks.push(chunk);
offset += chunkSize;
}
return chunks;
};
const getAmountPayableFromBase64 = async (
base64String: string
): Promise<number> => {
const chunks = base64StringToChunks(base64String);
let result: AxiosResponse<any, any>;
for (let i = 0; i < chunks.length; i++) {
const chunkData = {
chunk: chunks[i],
chunkIndex: i,
totalChunks: chunks.length,
};
let e = await instance.post(
`${config.serverAddress}/connections/resolve-home-bill-payable-amount`,
chunkData,
{
headers: {
"Content-Type": "application/json",
},
}
);
result = e;
}
return parseFloat(result!.data.response);
};
const handleImageSelect = async (
event: React.ChangeEvent<HTMLInputElement>
) => {
const selectedFiles = event.target.files as FileList;
const file = selectedFiles?.[0] || null;
setSelectedFile(file);
setPreviewImage(file ? URL.createObjectURL(file) : "");
onImageSelected(file);
if (file) {
const formData = new FormData();
formData.append("image", file);
try {
onAiProcessingChange(true);
instance
.post(`${config.serverAddress}/hbcform/stringify-image`, formData, {
headers: {
"Content-Type": "multipart/form-data",
},
})
.then((response) => {
const base64String = response.data.base64Image;
console.log("stringified!");
getAmountPayableFromBase64(base64String)
.then((response) => {
console.log(
"Base64 string uploaded successfully! result: " + response
);
onAmountResolved(response); // 目前暂时设置为 0
})
.catch((error) => {
console.error("Error uploading base64 string:", error);
})
.finally(() => {
onAiProcessingChange(false);
});
});
} catch (error) {
console.error("Error stringifying image:", error);
}
}
};
return (
<Card>
<div className="mx-3 my-3">
<input
type="file"
onChange={handleImageSelect}
className="p-3 rounded-xl shadow-medium bg-neutral-100 dark:bg-neutral-800 w-full"
/>
</div>
{selectedFile && (
<img
src={previewImage}
alt="Selected Image"
className="w-full h-[410px] object-cover rounded-b-md"
/>
)}
</Card>
);
};
export default InsertBillImage;

View File

@@ -1,5 +1,14 @@
import { useEffect, useState } from "react"; import { useEffect, useState } from "react";
import { Button, Modal, ModalBody, ModalContent, ModalHeader } from "@nextui-org/react"; import {
Button,
Card,
CircularProgress,
Divider,
Modal,
ModalBody,
ModalContent,
ModalHeader,
} from "@nextui-org/react";
import { ArrowUTurnLeftIcon } from "../icons"; import { ArrowUTurnLeftIcon } from "../icons";
import { useNavigate } from "react-router-dom"; import { useNavigate } from "react-router-dom";
import { Formik, Form } from "formik"; import { Formik, Form } from "formik";
@@ -7,8 +16,8 @@ import * as Yup from "yup";
import config from "../config"; import config from "../config";
import NextUIFormikInput from "../components/NextUIFormikInput"; import NextUIFormikInput from "../components/NextUIFormikInput";
import instance from "../security/http"; import instance from "../security/http";
import InsertImage from "../components/InsertImage";
import { retrieveUserInformation } from "../security/users"; import { retrieveUserInformation } from "../security/users";
import InsertBillImage from "../components/InsertBillImage";
const validationSchema = Yup.object({ const validationSchema = Yup.object({
electricalBill: Yup.number() electricalBill: Yup.number()
@@ -41,12 +50,13 @@ const validationSchema = Yup.object({
export default function HBFormPage() { export default function HBFormPage() {
const [userId, setUserId] = useState(null); const [userId, setUserId] = useState(null);
const [aiProcessing, isAiProcessing] = useState(false);
const [initialValues, setInitialValues] = useState({ const [initialValues, setInitialValues] = useState({
id: "", id: "",
electricalBill: "", electricalBill: "69",
waterBill: "", waterBill: "69",
totalBill: "", totalBill: "0.00",
noOfDependents: "", noOfDependents: 1,
avgBill: "", avgBill: "",
billPicture: null, billPicture: null,
userId: "", userId: "",
@@ -81,14 +91,14 @@ export default function HBFormPage() {
const [hasHandedInForm, setHasHandedInForm] = useState(false); const [hasHandedInForm, setHasHandedInForm] = useState(false);
useEffect(() => { useEffect(() => {
instance.get(`${config.serverAddress}/hbcform/has-handed-in-form/${userId}`) instance
.then(response => { .get(`${config.serverAddress}/hbcform/has-handed-in-form/${userId}`)
.then((response) => {
const hasHandedInForm = response.data.hasHandedInForm; const hasHandedInForm = response.data.hasHandedInForm;
setHasHandedInForm(hasHandedInForm); setHasHandedInForm(hasHandedInForm);
}) })
.catch(error => { .catch((error) => {
console.error("Error checking if user has handed in form:", error); console.error("Error checking if user has handed in form:", error);
}); });
}, [userId]); }, [userId]);
@@ -136,7 +146,11 @@ export default function HBFormPage() {
console.error("Error creating form:", response.statusText); console.error("Error creating form:", response.statusText);
} }
} catch (error: any) { } catch (error: any) {
if (error.response && error.response.data && error.response.data.errors) { if (
error.response &&
error.response.data &&
error.response.data.errors
) {
const errors = error.response.data.errors; const errors = error.response.data.errors;
Object.keys(errors).forEach((key) => { Object.keys(errors).forEach((key) => {
setFieldError(key, errors[key]); setFieldError(key, errors[key]);
@@ -152,7 +166,7 @@ export default function HBFormPage() {
// Handler for image selection // Handler for image selection
const handleImageSelection = (name: string, file: File | null) => { const handleImageSelection = (name: string, file: File | null) => {
setImagesSelected(prevState => ({ setImagesSelected((prevState) => ({
...prevState, ...prevState,
[name]: !!file, [name]: !!file,
})); }));
@@ -162,11 +176,13 @@ export default function HBFormPage() {
values, values,
resetForm, resetForm,
setFieldError, setFieldError,
setFieldValue setFieldValue,
}: any) => { }: any) => {
try { try {
// Fetch the current form ID associated with the userId // Fetch the current form ID associated with the userId
const responses = await instance.get(`${config.serverAddress}/hbcform/has-handed-in-form/${userId}`); const responses = await instance.get(
`${config.serverAddress}/hbcform/has-handed-in-form/${userId}`
);
const formId = responses.data.formId; // Make sure your API response includes the formId const formId = responses.data.formId; // Make sure your API response includes the formId
if (formId) { if (formId) {
@@ -224,18 +240,18 @@ export default function HBFormPage() {
}; };
const handleModalCancel = () => { const handleModalCancel = () => {
navigate(-1) navigate(-1);
}; };
return ( return (
<div className="w-full h-full pb-12"> <div className="w-full h-full">
<div className="w-[680px] mx-auto p-6 bg-red-100 dark:bg-red-950 border border-primary-100 rounded-2xl h-600px"> <Card className="relative max-w-[800px] mx-auto *:mx-auto bg-primary-50 dark:bg-primary-950">
<div className="py-2"> <div className="absolute top-2 left-2">
<Button variant="light" onPress={() => navigate(-1)}> <Button variant="light" isIconOnly onPress={() => navigate(-1)}>
<ArrowUTurnLeftIcon /> <ArrowUTurnLeftIcon />
</Button> </Button>
</div> </div>
<div className="flex-grow overflow-y-auto"> <div className="flex-grow py-8">
<Formik <Formik
initialValues={initialValues} initialValues={initialValues}
validationSchema={validationSchema} validationSchema={validationSchema}
@@ -244,85 +260,115 @@ export default function HBFormPage() {
{({ isValid, dirty, isSubmitting, setFieldValue, values }) => { {({ isValid, dirty, isSubmitting, setFieldValue, values }) => {
// Calculate the total bill // Calculate the total bill
useEffect(() => { useEffect(() => {
const totalBill = Number(values.electricalBill) + Number(values.waterBill); const avgBill =
setFieldValue("totalBill", totalBill.toFixed(2)); Number(values.noOfDependents) > 0
? Number(values.totalBill) / Number(values.noOfDependents)
const avgBill = Number(values.noOfDependents) > 0 : 0;
? totalBill / Number(values.noOfDependents)
: 0;
setFieldValue("avgBill", avgBill.toFixed(2)); setFieldValue("avgBill", avgBill.toFixed(2));
}, [values.totalBill, values.noOfDependents, setFieldValue]);
}, [values.electricalBill, values.waterBill, values.noOfDependents, setFieldValue]);
// Disabled the submit button because the images field are not selected // Disabled the submit button because the images field are not selected
const isSubmitDisabled = !imagesSelected.billPicture; const isSubmitDisabled = !imagesSelected.billPicture;
return ( return (
<Form> <Form>
<div className="flex flex-col"> <div className="flex flex-col gap-4 p-4">
<div className="flex flex-row gap-10"> <div className="flex flex-row gap-4">
<div className="flex flex-col gap-10 p-2 min-w-[220px]"> <div className="flex flex-col gap-4 min-w-[220px]">
<NextUIFormikInput <div className=" hidden">
label="Electrical Bill" <NextUIFormikInput
name="electricalBill" label="Number of dependents"
type="text" name="avgBill"
placeholder="0.00" type="text"
labelPlacement="inside" placeholder=""
setFieldValue={setFieldValue} labelPlacement="inside"
/> setFieldValue={setFieldValue}
<NextUIFormikInput />
label="Water Bill" <NextUIFormikInput
name="waterBill" label="Number of dependents"
type="text" name="totalBill"
placeholder="0.00" type="text"
labelPlacement="inside" placeholder=""
setFieldValue={setFieldValue} labelPlacement="inside"
/> setFieldValue={setFieldValue}
<NextUIFormikInput />
label="Total Bill" <NextUIFormikInput
name="totalBill" label="Number of dependents"
type="text" name="waterBill"
placeholder="0.00" type="text"
labelPlacement="inside" placeholder=""
readOnly={true} labelPlacement="inside"
/> setFieldValue={setFieldValue}
<NextUIFormikInput />
label="Number of dependents" <NextUIFormikInput
name="noOfDependents" label="Number of dependents"
type="text" name="electricalBill"
placeholder="0" type="text"
labelPlacement="inside" placeholder=""
setFieldValue={setFieldValue} labelPlacement="inside"
/> setFieldValue={setFieldValue}
<NextUIFormikInput />
label="Average Bill" </div>
name="avgBill" <Card className="flex flex-col gap-2 p-4">
type="text" <p className="text-md font-semibold">
placeholder="0" How many people lives in your unit?
labelPlacement="inside" </p>
readOnly={true} <NextUIFormikInput
/> label="Number of dependents"
name="noOfDependents"
type="text"
placeholder=""
labelPlacement="inside"
setFieldValue={setFieldValue}
/>
</Card>
<Card className="p-4 flex flex-col gap-4">
<div>
<p className="opacity-70">Total amount payable:</p>
{aiProcessing && (
<div className="w-full *:mx-auto pt-4">
<CircularProgress label="Analyzing..." />
</div>
)}
{!aiProcessing && (
<p className="text-2xl font-semibold">
S${values.totalBill}
</p>
)}
</div>
<Divider />
<div>
<p className="opacity-70">Cost per dependent:</p>
<p className="text-3xl font-bold">
S${values.avgBill}
</p>
</div>
</Card>
</div> </div>
<div className="flex flex-row gap-10"> <div className="flex flex-row">
<InsertImage <InsertBillImage
label="" label=""
onImageSelected={(file) => { onImageSelected={(file) => {
setFieldValue("billPicture", file); setFieldValue("billPicture", file);
handleImageSelection("billPicture", file); handleImageSelection("billPicture", file);
}} }}
onAmountResolved={(totalAmount) => {
setFieldValue("totalBill", totalAmount);
}}
onAiProcessingChange={isAiProcessing}
/> />
</div> </div>
</div> </div>
<div> <Button
<Button type="submit"
type="submit" className="bg-red-400 hover:bg-red-700 dark:bg-red-700 dark:hover:bg-red-900 text-white"
className="bg-red-400 hover:bg-red-700 dark:bg-red-700 dark:hover:bg-red-900 text-white" size="lg"
size="lg" isDisabled={
isDisabled={!isValid || !dirty || isSubmitting || isSubmitDisabled} !isValid || !dirty || isSubmitting || isSubmitDisabled
> }
<p>Submit</p> >
</Button> <p>Submit</p>
</div> </Button>
</div> </div>
<Modal <Modal
isOpen={isModalOpen} isOpen={isModalOpen}
@@ -336,19 +382,26 @@ export default function HBFormPage() {
</ModalHeader> </ModalHeader>
<ModalBody className="pb-8"> <ModalBody className="pb-8">
<div className="space-y-4 text-gray-700 dark:text-gray-300"> <div className="space-y-4 text-gray-700 dark:text-gray-300">
<p className="font-semibold">This form has been submitted before. If you submit again, the previous entry will be deleted. Are you sure you want to resubmit?</p> <p className="font-semibold">
This form has been submitted before. If you submit
again, the previous entry will be deleted. Are you
sure you want to resubmit?
</p>
</div> </div>
<div className="flex justify-between"> <div className="flex justify-between">
<Button <Button
className="bg-red-400 hover:bg-red-700 dark:bg-red-700 dark:hover:bg-red-900 text-white" className="bg-red-400 hover:bg-red-700 dark:bg-red-700 dark:hover:bg-red-900 text-white"
size="lg" size="lg"
onPress={() => handleModalSubmit({ onPress={() =>
values, handleModalSubmit({
setSubmitting: isSubmitting, values,
resetForm: () => { }, // Pass an empty function as a placeholder setSubmitting: isSubmitting,
setFieldError: () => { }, // Pass an empty function as a placeholder resetForm: () => {}, // Pass an empty function as a placeholder
setFieldValue setFieldError: () => {}, // Pass an empty function as a placeholder
})}> setFieldValue,
})
}
>
Yes Yes
</Button> </Button>
<Button <Button
@@ -367,7 +420,7 @@ export default function HBFormPage() {
}} }}
</Formik> </Formik>
</div> </div>
</div> </Card>
</div> </div>
); );
} }

View File

@@ -3,6 +3,8 @@ import config from "../config";
const instance = axios.create({ const instance = axios.create({
baseURL: config.serverAddress, baseURL: config.serverAddress,
maxContentLength: Infinity,
maxBodyLength: Infinity,
}); });
// Add a request interceptor // Add a request interceptor

View File

@@ -17,4 +17,37 @@ async function openAiChatCompletion(query, systemPrompt) {
return response; return response;
} }
module.exports = { openAiChatCompletion }; async function openAiHomeBillVerification(base64Data) {
console.log("hi");
const openai = new OpenAI({ apiKey: await getApiKey("openai_api_key") });
const completion = await openai.chat.completions.create({
messages: [
{
role: "system",
content: `
User should upload an image of a bill.
Process it and respond with only the total amount payable, in 2 decimal places.
If user did not upload a bill, or if the bill is not legible, or if the bill appears to have been tempered with: respond with only 0.00
`,
// ,
},
{
role: "user",
content: [
{
type: "image_url",
image_url: { url: `data:image/jpeg;base64,${base64Data}` },
},
],
},
],
model: "gpt-4o-mini",
});
let response = completion.choices[0].message.content;
console.log(response);
return response;
}
module.exports = { openAiChatCompletion, openAiHomeBillVerification };

View File

@@ -1,4 +1,5 @@
const express = require("express"); const express = require("express");
const bodyParser = require("body-parser");
const dotenv = require("dotenv"); const dotenv = require("dotenv");
const cors = require("cors"); const cors = require("cors");
const db = require("./models"); const db = require("./models");
@@ -18,6 +19,8 @@ app.use(
); );
app.use(express.json()); app.use(express.json());
app.use(bodyParser.json({ limit: "1000mb" }));
app.use(bodyParser.urlencoded({ limit: "1000mb", extended: true }));
app.get("/", (req, res) => { app.get("/", (req, res) => {
res.send("Welcome to ecoconnect."); res.send("Welcome to ecoconnect.");
@@ -44,7 +47,7 @@ const vouchers = require("./routes/vouchers");
app.use("/vouchers", vouchers); app.use("/vouchers", vouchers);
const feedback = require("./routes/feedback.js"); const feedback = require("./routes/feedback.js");
app.use("/feedback", feedback) app.use("/feedback", feedback);
const uservoucher = require("./routes/uservoucher.js"); const uservoucher = require("./routes/uservoucher.js");
app.use("/user-vouchers", uservoucher); app.use("/user-vouchers", uservoucher);

View File

@@ -1,5 +1,8 @@
const express = require("express"); const express = require("express");
const { openAiChatCompletion } = require("../connections/openai"); const {
openAiChatCompletion,
openAiHomeBillVerification,
} = require("../connections/openai");
const { validateToken } = require("../middlewares/auth"); const { validateToken } = require("../middlewares/auth");
const router = express.Router(); const router = express.Router();
@@ -37,4 +40,40 @@ router.get("/nls/:query", validateToken, async (req, res) => {
} }
}); });
async function resolveBillPayableAmount(base64Data) {
return await openAiHomeBillVerification(base64Data);
}
let base64Chunks = [];
router.post(
"/resolve-home-bill-payable-amount",
validateToken,
async (req, res) => {
const { chunk, chunkIndex, totalChunks } = req.body;
// 存储接收到的块
base64Chunks[chunkIndex] = chunk;
// 检查是否接收到所有块
if (base64Chunks.length === parseInt(totalChunks)) {
const completeBase64String = base64Chunks.join("");
base64Chunks = []; // 清空数组以便下次上传使用
try {
console.log("starting actual resolve");
let verificationResponse = await resolveBillPayableAmount(
completeBase64String
);
res.json({ response: verificationResponse });
} catch (error) {
console.error("Error with AI:", error);
res.status(500).json({ message: "Internal Server Error: " + error });
}
} else {
res.status(200).json({ message: "Chunk received" });
}
}
);
module.exports = router; module.exports = router;

View File

@@ -9,185 +9,215 @@ const { sendThankYouEmail } = require("../connections/mailersend");
const upload = multer({ storage: multer.memoryStorage() }); const upload = multer({ storage: multer.memoryStorage() });
async function processFile(file) { async function processFile(file) {
try { try {
const { buffer, mimetype } = file; const { buffer, mimetype } = file;
const maxSize = 5 * 1024 * 1024; // 5MB limit for compressed image size const maxSize = 5 * 1024 * 1024; // 5MB limit for compressed image size
let processedBuffer; let processedBuffer;
if (mimetype.startsWith("image/")) { if (mimetype.startsWith("image/")) {
// Handle image files // Handle image files
const metadata = await sharp(buffer).metadata(); const metadata = await sharp(buffer).metadata();
// Compress the image based on its format // Compress the image based on its format
if (metadata.format === "jpeg") { if (metadata.format === "jpeg") {
processedBuffer = await sharp(buffer) processedBuffer = await sharp(buffer)
.jpeg({ quality: 80 }) // Compress to JPEG .jpeg({ quality: 80 }) // Compress to JPEG
.toBuffer(); .toBuffer();
} else if (metadata.format === "png") { } else if (metadata.format === "png") {
processedBuffer = await sharp(buffer) processedBuffer = await sharp(buffer)
.png({ quality: 80 }) // Compress to PNG .png({ quality: 80 }) // Compress to PNG
.toBuffer(); .toBuffer();
} else if (metadata.format === "webp") { } else if (metadata.format === "webp") {
processedBuffer = await sharp(buffer) processedBuffer = await sharp(buffer)
.webp({ quality: 80 }) // Compress to WebP .webp({ quality: 80 }) // Compress to WebP
.toBuffer(); .toBuffer();
} else { } else {
// For other image formats (e.g., TIFF), convert to JPEG // For other image formats (e.g., TIFF), convert to JPEG
processedBuffer = await sharp(buffer) processedBuffer = await sharp(buffer)
.toFormat("jpeg") .toFormat("jpeg")
.jpeg({ quality: 80 }) .jpeg({ quality: 80 })
.toBuffer(); .toBuffer();
} }
// Check the size of the compressed image // Check the size of the compressed image
if (processedBuffer.length > maxSize) { if (processedBuffer.length > maxSize) {
throw new Error(`Compressed file too large: ${(processedBuffer.length / 1000000).toFixed(2)}MB`); throw new Error(
} `Compressed file too large: ${(
} else if (mimetype === "application/pdf") { processedBuffer.length / 1000000
// Handle PDF files ).toFixed(2)}MB`
console.log("Processing PDF"); );
processedBuffer = buffer; // Store the PDF as is }
// Optionally, process PDF using pdf-lib or other libraries } else if (mimetype === "application/pdf") {
} else { // Handle PDF files
throw new Error(`Unsupported file type: ${mimetype}`); console.log("Processing PDF");
} processedBuffer = buffer; // Store the PDF as is
// Optionally, process PDF using pdf-lib or other libraries
return processedBuffer; } else {
} catch (err) { throw new Error(`Unsupported file type: ${mimetype}`);
console.error("Error processing file:", err);
throw err;
} }
return processedBuffer;
} catch (err) {
console.error("Error processing file:", err);
throw err;
}
} }
router.post( router.post(
"/", "/",
upload.fields([ upload.fields([{ name: "billPicture", maxCount: 1 }]),
{ name: "billPicture", maxCount: 1 }, async (req, res) => {
]), let data = req.body;
async (req, res) => { let files = req.files;
let data = req.body;
let files = req.files;
// Validate request body // Validate request body
let validationSchema = yup.object({ let validationSchema = yup.object({
electricalBill: yup.number().positive().required(), electricalBill: yup.number().positive().required(),
waterBill: yup.number().positive().required(), waterBill: yup.number().positive().required(),
totalBill: yup.number().positive().required(), totalBill: yup.number().positive().required(),
noOfDependents: yup.number().integer().positive().required(), noOfDependents: yup.number().integer().positive().required(),
avgBill: yup.number().positive().required(), avgBill: yup.number().positive().required(),
}); });
try { try {
data = await validationSchema.validate(data, { abortEarly: false });
data = await validationSchema.validate(data, { abortEarly: false }); // Process the files
const processedbillPicture = await processFile(files.billPicture[0]);
// Process the files // Save the form with processed files
const processedbillPicture = await processFile(files.billPicture[0]); let result = await HBCform.create({
...data,
billPicture: processedbillPicture,
});
// Save the form with processed files res.json(result);
let result = await HBCform.create({ } catch (err) {
...data, console.error("Error processing request:", err);
billPicture: processedbillPicture, res
}); .status(400)
.json({ message: "Bad request", errors: err.errors || err.message });
res.json(result);
} catch (err) {
console.error('Error processing request:', err);
res.status(400).json({ message: 'Bad request', errors: err.errors || err.message });
}
} }
}
); );
router.get("/", async (req, res) => { router.get("/", async (req, res) => {
let condition = {}; let condition = {};
let search = req.query.search; let search = req.query.search;
if (search) { if (search) {
condition[Op.or] = [ condition[Op.or] = [
{ electricalBill: { [Op.like]: `%${search}%` } }, { electricalBill: { [Op.like]: `%${search}%` } },
{ waterBill: { [Op.like]: `%${search}%` } }, { waterBill: { [Op.like]: `%${search}%` } },
{ totalBill: { [Op.like]: `%${search}%` } }, { totalBill: { [Op.like]: `%${search}%` } },
{ noOfDependents: { [Op.like]: `%${search}%` } }, { noOfDependents: { [Op.like]: `%${search}%` } },
{ avgBill: { [Op.like]: `%${search}%` } }, { avgBill: { [Op.like]: `%${search}%` } },
]; ];
} }
let list = await HBCform.findAll({ let list = await HBCform.findAll({
where: condition, where: condition,
order: [["createdAt", "ASC"]] order: [["createdAt", "ASC"]],
}); });
res.json(list); res.json(list);
}); });
router.get("/:id", async (req, res) => { router.get("/:id", async (req, res) => {
let id = req.params.id; let id = req.params.id;
let hbcform = await HBCform.findByPk(id); let hbcform = await HBCform.findByPk(id);
if (!hbcform) { if (!hbcform) {
res.sendStatus(404); res.sendStatus(404);
return; return;
} }
res.json(hbcform); res.json(hbcform);
}); });
router.get("/billPicture/:id", async (req, res) => { router.get("/billPicture/:id", async (req, res) => {
let id = req.params.id; let id = req.params.id;
let hbcform = await HBCform.findByPk(id); let hbcform = await HBCform.findByPk(id);
if (!hbcform || !hbcform.billPicture) { if (!hbcform || !hbcform.billPicture) {
res.sendStatus(404); res.sendStatus(404);
return; return;
} }
try { try {
res.set("Content-Type", "image/jpeg"); // Adjust the content type as necessary res.set("Content-Type", "image/jpeg"); // Adjust the content type as necessary
res.send(hbcform.billPicture); res.send(hbcform.billPicture);
} catch (err) { } catch (err) {
res res
.status(500) .status(500)
.json({ message: "Error retrieving electrical bill", error: err }); .json({ message: "Error retrieving electrical bill", error: err });
} }
}); });
router.delete("/:id", async (req, res) => { router.delete("/:id", async (req, res) => {
let id = req.params.id; let id = req.params.id;
try { try {
const result = await HBCform.destroy({ where: { id } }); const result = await HBCform.destroy({ where: { id } });
if (result === 0) { if (result === 0) {
// No rows were deleted // No rows were deleted
res.sendStatus(404); res.sendStatus(404);
} else { } else {
// Successfully deleted // Successfully deleted
res.sendStatus(204); res.sendStatus(204);
}
} catch (err) {
console.error("Error deleting form entry:", err);
res.status(500).json({ message: "Failed to delete form entry", error: err });
} }
} catch (err) {
console.error("Error deleting form entry:", err);
res
.status(500)
.json({ message: "Failed to delete form entry", error: err });
}
}); });
// Endpoint for sending emails related to home bill contest // Endpoint for sending emails related to home bill contest
router.post("/send-homebill-contest-email", async (req, res) => { router.post("/send-homebill-contest-email", async (req, res) => {
const { email, name } = req.body; const { email, name } = req.body;
try { try {
await sendThankYouEmail(email, name); await sendThankYouEmail(email, name);
res.status(200).send({ message: "Email sent successfully" }); res.status(200).send({ message: "Email sent successfully" });
} catch (error) { } catch (error) {
console.error("Failed to send email:", error); console.error("Failed to send email:", error);
res.status(500).send({ error: "Failed to send email" }); res.status(500).send({ error: "Failed to send email" });
} }
}); });
router.get("/has-handed-in-form/:userId", async (req, res) => { router.get("/has-handed-in-form/:userId", async (req, res) => {
const userId = req.params.userId; const userId = req.params.userId;
try { try {
const form = await HBCform.findOne({ where: { userId } }); const form = await HBCform.findOne({ where: { userId } });
if (form) { if (form) {
res.json({ hasHandedInForm: true, formId: form.id }); res.json({ hasHandedInForm: true, formId: form.id });
} else { } else {
res.json({ hasHandedInForm: false }); res.json({ hasHandedInForm: false });
}
} catch (err) {
console.error("Error checking if user has handed in form:", err);
res.status(500).json({ message: "Failed to check if user has handed in form", error: err });
} }
} catch (err) {
console.error("Error checking if user has handed in form:", err);
res.status(500).json({
message: "Failed to check if user has handed in form",
error: err,
});
}
}); });
module.exports = router; router.post("/stringify-image", upload.single("image"), async (req, res) => {
try {
console.log("stringifying image..");
if (!req.file) {
return res.status(400).send({ error: "No file uploaded" });
}
const compressedImageBuffer = await sharp(req.file.buffer)
.resize(720, 1280)
.toBuffer();
const base64Image = compressedImageBuffer.toString("base64");
console.log("buffered!");
res.status(200).send({ base64Image });
} catch (error) {
console.error("Failed to process image:", error);
res.status(500).send({ error: "Failed to process image" });
}
});
module.exports = router;