Viewing File: /home/ubuntu/combine_ai/writing_aid/src/main.py

# -*- coding: utf-8 -*-

from dotenv import load_dotenv
import sys, os
import json
from src.constants import *
from src.utils import *
from src.prompt import *
import requests
import requests
import openai
from openai import OpenAI
# set_seed(42)

openai.api_key = os.environ["OPENAI_API_KEY"]
client = OpenAI()

def data_querying(user_id, description, use_case, language, tone, more_details, additional_info):

    # description,uuid,language,tone,more_details,additional_info

    if description == "/start":
        welcome_message = "Welcome to our service! I'm Liya, your friendly assistant. How may I help you today? Feel free to ask any questions."
        return welcome_message

    create_json(user_id)


    # Create Prompt
    prompt = create_prompt(description, use_case, language, tone, more_details, additional_info)
    print("prompt", prompt)
    
    system_prompt = {
        "role": "system",
        "content": prompt
    }
    user_query = {
        "role": "user",
        "content": description
    }

    model_input = [system_prompt, user_query]
    response = openai.chat.completions.create(model="gpt-3.5-turbo", messages=model_input)
    assistant_response = response.choices[0].message.content.strip()

    # Update User Memory
    if description != "\\start":
        save_response(assistant_response, user_id, description)
    # save_response(last_response, user_id, input_text)

    return assistant_response


def answer_question(question, user_id):
    user_id = "static_user_id"
    try:
        # response = data_querying(question)
        description = question
        use_case = "Cover Letter"
        language = "English"
        tone = None
        more_details = None
        additional_info = None
        response = data_querying(user_id, description, use_case, language, tone, more_details, additional_info)

        return response
    except Exception as e:
        return str(e)