added OpenAi controllers and configurations

Author: sanjaraiy

frontend/src/components/GeneratedImageCard.js

@@ -2,18 +2,19 @@ import { CircularProgress } from '@mui/material'
 import React from 'react'
 import styled from 'styled-components'
 
-const Container = styled.dev`
+const Container = styled.div`
 flex:1;
 padding: 16px;
 display: flex;
+border-radius: 20px;
+align-items: center;
+justify-content: center;
 min-height:300px;
 gap: 16px;
 flex-direction: column;
 border:2px dashed ${({theme}) => theme.yellow};
 color: ${({theme}) => theme.arrow + 80};
-border-radius: 20px;
-align-items: center;
-justify-content: center;
+
 `
 
 const Image = styled.img`

server/controllers/GenerateAIImage.controller.js

@@ -0,0 +1,29 @@
+import dotenv from "dotenv";
+import { createError } from "../utils/error";
+import {Configuration, OpenAIApi} from "openai";
+
+dotenv.config();
+
+
+const configuration = new Configuration({
+    apiKey: process.env.OPEN_AI_SECRET_KEY
+})
+
+const openai = new OpenAIApi(configuration);
+
+export const generateImage = async(req, res, next) => {
+try {
+     const {prompt} = req.body;
+
+     const response = await openai.createImage({
+        prompt,
+        n:1,
+        size: "1024x1024",
+        response_format: "b64_json"
+     })
+     const generateImage = response.data.data[0].b64_json;
+     return res.status(200).json({photo: generateImage});
+} catch (error) {
+    next(createError(error.status, error?.response?.data?.error?.message || error?.message));
+}
+}

server/controllers/Posts.controller.js

@@ -0,0 +1,35 @@
+import Post from "../models/Posts.model";
+import dotenv from "dotenv"
+import { createError } from "../utils/error";
+import { v2 as cloudinary } from 'cloudinary';
+
+cloudinary.config({ 
+    cloud_name: process.env.CLOUDINARY_CLOUD_NAME,
+    api_key: process.env.CLOUDINARY_API_KEY,
+    api_secret: process.env.CLOUDINARY_API_SECRET,
+});
+
+export const getAllPosts = async(req, res, next) => {
+    try {
+        const posts = await Post.find({});
+        return res.status(200).json({success: true, data: posts})
+    } catch (error) {
+        next(createError(error.status, error?.response?.data?.error?.message || error?.message));
+    }
+}
+
+export const createPost = async (req, res, next) => {
+    try {
+        const {name, prompt, photo} = req.body;
+        const photoUrl = await cloudinary.uploader.upload(photo);
+        const newPost = await Post.create({
+            name,
+            prompt,
+            photo: photoUrl?.secure_url,
+        });
+
+        return res.status(201).json({success:true, data: newPost});
+    } catch (error) {
+        next(createError(error.status, error?.response?.data?.error?.message || error?.message));
+    }
+}

server/models/Posts.model.js

@@ -0,0 +1,20 @@
+import mongoose, { Schema } from 'mongoose'
+
+const PostSchema = new Schema({
+    name:{
+        type: String,
+        required: true
+    },
+    prompt: {
+        type: String,
+        required: true,
+    },
+    photo: {
+        type: String,
+        required: true,
+    },
+},{timestamps:true});
+
+const Post = mongoose.model("Post", PostSchema);
+
+export default Post;

server/package-lock.json

@@ -5,8 +5,7 @@
   "type": "module",
   "main": "server.js",
   "scripts": {
-    "start":"nodemon server.js"
-   
+    "start": "nodemon server.js"
   },
   "author": "sanjay rai",
   "license": "ISC",
@@ -15,6 +14,7 @@
     "cors": "^2.8.5",
     "dotenv": "^16.4.5",
     "express": "^4.19.2",
+    "mongodb": "^6.8.0",
     "mongoose": "^8.4.5",
     "nodemon": "^3.1.4",
     "openai": "^4.52.3"

server/package.json

@@ -0,0 +1,10 @@
+import express from "express";
+import { createPost, getAllPosts } from "../controllers/Posts.controller";
+import { generateImage } from "../controllers/GenerateAIImage.controller";
+
+const router = express.Router();
+
+router.post("/", generateImage);
+
+export default router;
+

server/routes/GenerateAIImage.route.js

@@ -0,0 +1,9 @@
+import express from "express";
+import { createPost, getAllPosts } from "../controllers/Posts.controller";
+
+const router = express.Router();
+
+router.get("/", getAllPosts);
+router.post("/", createPost);
+export default router;
+

server/routes/Posts.route.js

@@ -2,11 +2,14 @@ import cors from "cors";
 import mongoose from "mongoose";
 import dotenv from "dotenv";
 import express from "express";
+import PostRouter from "./routes/Posts.route";
+import GenerateImageRouter from "./routes/GenerateAIImage.route";
 
 dotenv.config();
-const app = express();
 
+const app = express();
 const PORT = process.env.PORT || 8080;
+const URL = process.env.MONGODB_URI;
 
 //========== default middlewares ===================
 app.use(cors());
@@ -22,6 +25,9 @@ app.use((err,req,res,next)=>{
     }) 
 })
 
+
+app.use("/api/v1/post", PostRouter);
+app.use("/api/v1/generateImage",  GenerateImageRouter)
 //=========== Default route =============
 app.get("/", async (req,res) => {
     res.status(200).json({
@@ -30,9 +36,23 @@ app.get("/", async (req,res) => {
 })
 
 
+
+//============ Connect to mongodb ==============
+const connectDB = () => {
+    mongoose.set("strictQuery", true);
+    mongoose.connect(URL)
+    .then(() => {
+           console.log("Successfully Connected MongoDB...!!");
+    })
+    .catch((err) => {
+        console.error("Failed to Connect to DB");
+    })
+}
+
 //============ Server Running Function ===================
 const startServer = () =>{
     try {
+        connectDB()
         app.listen(PORT, () => {
             console.log("Server started on Port: ",PORT);
         })

server/server.js

@@ -0,0 +1,6 @@
+export const createError = (status, message) => {
+    const err = new Error();
+    err.status = status;
+    err.message = message;
+    return err;
+}