app.py

import uvicorn
import os
from fastapi import FastAPI, Request
from src.graphs.graph_builder import GraphBuilder
from src.llms.groqllm import GroqLLM
from dotenv import load_dotenv

load_dotenv()

app = FastAPI()

os.environ["LANGSMITH_API_KEY"] = os.getenv("LANGCHAIN_API_KEY")

@app.get("/")
async def root():
    return {"message": "Welcome to the Blog Generator API"}

@app.post("/blogs")
async def create_blog(request: Request):
    print("LLM object created successfully")
    data = await request.json()
    topic = data.get("topic", "")
    language = data.get("language", "")

    # get the llm object
    groqllm = GroqLLM()
    llm = groqllm.get_llm()
    # get the graph
    graph_builder = GraphBuilder(llm)
    if language and topic:
        graph = graph_builder.setup_graph(usecase="language")
        state = graph.invoke({"topic": topic, "current_language": language.lower()})
    elif topic:
        graph = graph_builder.setup_graph("topic")
        state = graph.invoke({"topic": topic})

    return {"data" : state}

if __name__ == "__main__":
    uvicorn.run("app:app", host = "0.0.0.0", port = 8000, reload = True)