erp-ai-latest/src/services/chatbot.py

import os
import json
from openai import OpenAI
from pydantic import BaseModel, Field
from typing import List, Dict, Optional
from src.prompts.sops import *
from src.prompts.chatbot import *
from src.models.sop_response_schemas import *
from src.models.bot_response_schema import *
from scripts.assessment_data import generate_summary_stats_v2
from dotenv import load_dotenv
load_dotenv()


#SopGeneratorDocument
class Chatbot:
    def __init__(self):
        self.api_key = os.getenv("OPENAI_API_KEY")
        self.client = OpenAI(api_key=self.api_key)
        self.model = "gpt-4o-mini"

    def _extract_text_from_docs(self, docs):
        """Extract text content from document objects."""
        return [doc.page_content for doc in docs]
    # Existing methods...

    def validate_worker(self, question, docs) -> VisionMissionResponse:
        """
        This method is responsible for validating the worker's response ("yes" or "no") using the provided document(s).
        The system generates a prompt, extracts document content, and validates the response using the GPT-4 model.

        :param question: The yes/no question asked to the worker.
        :param docs: A list of document(s) uploaded by the worker.
        :return: VisionMissionResponse containing the validated result or None if an error occurs.
        """
        try:
            docs_text = self._extract_text_from_docs(docs)
            prompt = validate_worker_prompt()
            response = self.client.beta.chat.completions.parse(
                model=self.model,
                messages=[
                    {
                        "role": "system",
                        "content": f'''{prompt} The document is uploaded next.'''
                    },
                    {
                        "role": "user",
                        "content": f"Question :{question}",
                    },
                    {
                        "role": "user",
                        "content": [{"type": "text", "text": text} for text in docs_text],
                    }
                ],
                response_format=ValidateWorker,
                max_tokens=1024,
                temperature=0.1
            )

            # Parse the response from the LLM
            extracted_text = json.loads(response.choices[0].message.content)

            return extracted_text

        except Exception as e:
            print(f"An error occurred: {e}")
            return None
        
    


    def predict_based_on_past_assessment(self, query, company_info, companyid) -> Result:
        """
        This method generates predictions based on past assessment data of a company. It queries the backend for the 
        company's assessment data, generates a prompt, and then uses the GPT-4 model to return predictions based on the query.

        :param query: The question or query asked by the user.
        :param company_info: General information about the company (name, size, departments, etc.).
        :param companyid: Unique identifier of the company to fetch its specific data.
        :return: Result containing the prediction result or None if an error occurs.
        """
        try:
            # Define the path to the company's assessment data (stored as a CSV)
            data_path = os.path.join('data', 'raw', 'erp_company_assessment', f'{companyid}_raw_data.csv')
            
            # Generate summary statistics from the company's assessment data
            summary_stats = generate_summary_stats_v2(file_path=data_path)
           

            # Generate the prompt using the company info and the summary statistics
            prompt = predict_based_past_assessment_prompt(
                query=query,
                company_info=company_info,
                summary_stats=summary_stats
            )
   
            # Interact with GPT-4 model to get a response
            response = self.client.beta.chat.completions.parse(
                model=self.model,
                messages=[
                    {
                        "role": "system",
                        "content": f"{prompt}"
                    },
                    {
                        "role": "user",
                        "content": f"{query}",
                    }
                ],
                response_format=Result,
                max_tokens=1024,
                temperature=0.1
            )

            # Extract and return the response from the GPT-4 model
            extracted_text = json.loads(response.choices[0].message.content)

            return extracted_text

        except Exception as e:
            print(f"An error occurred: {e}")
            return None

    
    def predict_next_n_assessment(self, company_info, companyid, N) -> AssessmentPredictionsResponse:
        """
        This method generates predictions based on past assessment data of a company. It queries the backend for the 
        company's assessment data, generates a prompt, and then uses the GPT-4 model to return predictions based on the query.

        :param query: The question or query asked by the user.
        :param company_info: General information about the company (name, size, departments, etc.).
        :param companyid: Unique identifier of the company to fetch its specific data.
        :param N: Number of assessments to predict.
        :return: Result containing the prediction result or None if an error occurs.
        """
        try:
            # Define the path to the company's assessment data (stored as a CSV)
            data_path = os.path.join('data', 'raw', 'erp_company_assessment', f'{companyid}_raw_data.csv')
            
            # Generate summary statistics from the company's assessment data
            summary_stats = generate_summary_stats_v2(file_path=data_path)

            # Generate the prompt using the company info and the summary statistics
            prompt = predict_next_n_assessments_prompt()
          
            
            # Interact with GPT-4 model to get a response
            response = self.client.beta.chat.completions.parse(
                model=self.model,
                messages=[
                    {
                        "role": "system",
                        "content": f"{prompt}"
                    },
                    {
                        "role": "user",
                        "content": f"company info: {company_info}--> N-value is {N} ",
                    },
                    {
                        "role": "user",
                        "content": f"Summary stats: {summary_stats}",
                    }
                ],
                response_format=AssessmentPredictionsResponse,
                max_tokens=1024,
                temperature=0.1
            )

            # Extract the response from the GPT-4 model
            extracted_text = json.loads(response.choices[0].message.content)

            # Initialize dictionary to store assessments with dynamic names
            predictions = {}

            # Loop through the predicted assessments and rename them dynamically
            for i in range(N):
                assessment_key = f"assessment_{i + 1}"
                predictions[assessment_key] = extracted_text["predictions"][i]['AssessmentN']

            # Return the dynamically named assessments
            return predictions

        except Exception as e:
            print(f"An error occurred: {e}")
            return None