from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
from utils import get_mongo_collection, groq_calls
from groq import Groq
from dotenv import load_dotenv
import os
import random
import uvicorn
from bson import ObjectId

load_dotenv()



app = FastAPI(title="Mind Dump Summariser")

class ReportRequest(BaseModel):
    user_id: str
    username: str



@app.post("/cron_schedulerbot")
async def cron_test():
    return "Hello Cron Tester"



@app.post("/mindDump")
async def generate_report(request: ReportRequest):
    try:
        userid = request.user_id   
        username = request.username     
        journalcollection = get_mongo_collection()
        cursor = journalcollection.find({"type":"mind_dump","userId":userid}).sort("createdAt",-1).limit(1)
        original = ""
        for item in cursor:
            if item["content"].get('original'):
                # print(item["content"]["original"])
                original = item["content"].get('original')

        print(original)
        
        prompt = f"""
        I am giving you the complete mind dump as written by the user :{original}
        You can greet and address him with his name: {username}
        Now you are working as a keen oberserver who can identify patterns among a series of entangled thoughts
        SO basically there are multiple knots that will be present inside the user's mind. ALl of that he has written in the mind dump
        Now starting from start to finish and linking multiple parts in proper series you have to tell user what he has been expriencing
        You need to untie those knots identifying the problems adn what could possibly be causing them
        You should not bring rememdies or solution externally, whatever you say to the user has to be from what he has written in the mind dump section
        Remember, you're not being asked as a chatbot, you'll have to run just one time and hence whatever output you give should be complete in it's own sense
        You shouldn't ask questions or further clarifications.
 
        """

        final = groq_calls(prompt)

        return {"Summary points": final}

    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))
    
if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=8000)