aiEvaluateImageToStar

pull/7/head
Kar at l5 2024-12-21 15:26:13 +05:30
parent e520169fb9
commit b7b168b2d0
1 changed files with 8 additions and 10 deletions

View File

@ -1,7 +1,7 @@
const AWS = require('aws-sdk'); const AWS = require('aws-sdk');
const axios = require('axios'); const axios = require('axios');
const mongoose = require('mongoose'); const mongoose = require('mongoose');
const Game = require('../../models/gameModel'); // Ensure this path matches your project structure const Game = require('../../models/gameModel');
const aiEvaluateImageToStar = async (req, res) => { const aiEvaluateImageToStar = async (req, res) => {
try { try {
@ -32,7 +32,7 @@ const aiEvaluateImageToStar = async (req, res) => {
content: [ content: [
{ {
type: 'text', type: 'text',
text: 'The coloring was done by a 5-year-old kid. Give a score between 1 to 10 in JSON format.', text: 'The coloring was done by a 5-year-old kid. Give a score between 3 to 5 in JSON format.',
}, },
{ {
type: 'image_url', type: 'image_url',
@ -57,20 +57,18 @@ const aiEvaluateImageToStar = async (req, res) => {
}, },
} }
); );
const rResp=openAiResponse.data.choices[0].message;
// Extract the score from OpenAI's response const cleanContent = rResp.content.replace(/```json|```/g, '').trim();
const scoreResponse = openAiResponse.data.choices[0].message.content; const parsedScore = JSON.parse(cleanContent);
const parsedScore = JSON.parse(scoreResponse); // console.log(parsedScore); // { score: 8 }
console.log(openAiResponse)
const gameData = new Game({ const gameData = new Game({
gameName, gameName,
userId, userId,
gameID, gameID,
gameTime, gameTime,
score: parsedScore.score || 'N/A', // Adjust key to match OpenAI response score: parsedScore.score || 'N/A',
gameStar, gameStar,
screenshotUrl, screenshotUrl:screenshotUrl.data.filePath,
}); });
await gameData.save(); await gameData.save();