Backend

app.py

@@ -1,6 +1,3 @@
-from fastapi import FastAPI
-from pydantic import BaseModel
-from dotenv import load_dotenv
 import faiss
 import numpy as np
 import pickle
@@ -8,13 +5,36 @@ import os
 from optimum.onnxruntime import ORTModelForFeatureExtraction
 from transformers import AutoTokenizer
 from groq import Groq
+from fastapi import FastAPI, Body
+from fastapi.responses import JSONResponse
+from pydantic import BaseModel
+from pymongo import AsyncMongoClient
+from bson import ObjectId
+from dotenv import load_dotenv
+import requests
+
+load_dotenv()
 
 app = FastAPI()
-class Ques(BaseModel):
-    question: str
+uri = os.getenv('MONGO_URI')
+client = AsyncMongoClient(uri)
 
+class Blogs(BaseModel):
+    timeToRead : str
+    blogDate : str
+    title : str
+    content : str
+    author : str
 
 
+    class Config:
+        json_encoders = {
+            ObjectId : str
+        }
+
+class Ques(BaseModel):
+    question: str
+
 
 class MinimalEmbedding:
     """Lightweight embedding model using ONNX Runtime (no PyTorch)"""
@@ -100,6 +120,7 @@ Cite the sources with [Page X, Section Y]."""
 
 embedding_model = MinimalEmbedding()
 chunks, faiss_index = embed_and_cache_chunks([])
+
 load_dotenv()
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 if not GROQ_API_KEY:
@@ -109,19 +130,147 @@ llm = Groq(
     api_key=GROQ_API_KEY
 )
 
+try:
+    database = client.get_database("MindfulNutrition")
+    blogs = database.get_collection("blogs")
+except Exception as e:
+    raise Exception("Exception: ",e)
+
+
 
 
 
-@app.get("/")
-def greet_json():
-    return {"Hello": "World!"}
+@app.get('/')
+def index():
+    return JSONResponse(
+        status_code=200,
+        content={
+            "data":"server is running"
+        }
+    )
+    # return {"data": "server is running"}
 
+# RAG
+@app.post('/api/chat')
+def chatResponse(body:Ques):
+    if body.question.strip() != '':
+        # LLM RAG code here 
+        user_query = body.question
+        retrieved_chunks = retrieve_chunks(user_query, chunks, faiss_index)
+        final_prompt = build_prompt(user_query, retrieved_chunks)
+        response = llm.chat.completions.create(model="llama-3.1-8b-instant", messages=[{'role':"user","content":final_prompt}])
+
+
+        return JSONResponse(
+            status_code=200,
+            content={"success":True,"data":response.choices[0].message.content}
+        )
+    else:
+        return JSONResponse(
+            status_code=400,
+            content={"success":False,"data":"Question cannot be empty"}
+        )
 
-@app.post('/chat')
-def runRag(body: Ques):
-    user_query = body.question
-    retrieved_chunks = retrieve_chunks(user_query, chunks, faiss_index)
-    final_prompt = build_prompt(user_query, retrieved_chunks)
-    response = llm.chat.completions.create(model="llama-3.1-8b-instant", messages=[{'role':"user","content":final_prompt}])
 
-    return {"data":response.choices[0].message.content}
+# Blogs
+@app.post('/api/createBlog')
+async def getBlog(body:Blogs):
+
+
+    result = await blogs.insert_one(body.model_dump())
+
+    # return {"data":body.model_dump()}
+    if result.inserted_id:
+        return JSONResponse(
+            status_code=200,
+            content={
+                "data":"Blog created successfully!"
+            }
+        )
+    else:
+        return JSONResponse(
+            status_code=400,
+            content={
+                "data":"An error occured while creating the blog"
+            }
+        )
+
+@app.get('/api/printBlogs')
+async def get_all_blogs():
+    cursor = blogs.find()
+    blogsList = []
+    async for blog in cursor:
+        # print(type(blog))
+        blog['_id'] = str(blog['_id'])
+        # blog_dict = BlogsResponse(**blog)
+        blogsList.append(blog)
+
+    # return {'data':blogsList}
+    return JSONResponse(
+        status_code=200,
+        content={
+            "data":blogsList
+        }
+    )
+
+@app.get('/api/Blogs/{id}')
+async def get_specific_blog(id:str):
+    try:
+        blog = blogs.find_one({"_id": ObjectId(id)})
+        blog['_id'] = str(blog['_id'])
+        return JSONResponse(
+            status_code=200,
+            content={
+                "data": blog
+                }
+            )
+
+    except:
+        return JSONResponse(
+            status_code=200,
+            content={
+                "data": "blog not found"
+                }
+            )
+        
+
+
+
+@app.patch('/api/editBlog')
+# Json
+# {_id':'whtv','setter':{multiple or single updating jsons}}
+async def editBlog(body:dict):
+    result = await blogs.update_one({'_id':ObjectId(body['_id'])},{'$set': body['setter']})
+    if result.matched_count == 0:
+        return JSONResponse(
+            status_code=400,
+            content={
+                "data": "Blog Not Found"
+            }
+        )
+    else:
+        return JSONResponse(
+            status_code=200,
+            content={
+                "data": "Blog updated successfully!"
+            }
+        )
+
+
+@app.delete('/api/deleteBlog')
+async def deleteBlog(body:dict):
+    result = await blogs.delete_one({'_id':ObjectId(body['_id'])})
+    if result.deleted_count>0:
+        return JSONResponse(
+            status_code=200,
+            content={
+                "data": "Blog deleted successfully!"
+            }
+        )
+    else:
+        return JSONResponse(
+            status_code=400,
+            content={
+                "data": "Blog was not found or was not deleted!"
+            }
+        )

requirements.txt

@@ -8,4 +8,6 @@ onnxruntime
 onnx
 optimum[onnxruntime]
 groq
-packaging
+packaging
+pymongo
+requests