mirror of
https://github.com/OMGeeky/gpt-pilot.git
synced 2026-01-01 17:09:59 +01:00
46 lines
1.7 KiB
Python
46 lines
1.7 KiB
Python
# user_tasks.py
|
|
import json
|
|
from termcolor import colored
|
|
|
|
from utils.utils import execute_step, split_into_bullets, find_role_from_step, generate_app_data
|
|
from database.database import save_progress, get_progress_steps
|
|
from logger.logger import logger
|
|
from prompts.prompts import get_additional_info_from_user, execute_chat_prompt
|
|
|
|
|
|
def get_user_tasks(summary, args):
|
|
current_step = 'user_tasks'
|
|
role = find_role_from_step(current_step)
|
|
# If this app_id already did this step, just get all data from DB and don't ask user again
|
|
steps = get_progress_steps(args['app_id'], current_step)
|
|
if steps and not execute_step(args['step'], current_step):
|
|
first_step = steps[0]
|
|
data = json.loads(first_step['data'])
|
|
|
|
summary = data.get('summary')
|
|
app_data = data.get('app_data')
|
|
if app_data is not None:
|
|
args.update(app_data)
|
|
|
|
message = f"User tasks already done for this app_id: {args['app_id']}. Moving to next step..."
|
|
print(colored(message, "green"))
|
|
logger.info(message)
|
|
return summary
|
|
|
|
# USER TASKS
|
|
print(colored(f"Generating user tasks...\n", "green"))
|
|
logger.info(f"Generating user tasks...")
|
|
|
|
user_tasks, user_tasks_messages = execute_chat_prompt('user_tasks/specs.prompt',
|
|
{'prompt': summary, 'app_type': args['app_type']},
|
|
current_step)
|
|
|
|
logger.info(split_into_bullets(user_tasks))
|
|
user_tasks = get_additional_info_from_user(split_into_bullets(user_tasks), role)
|
|
|
|
logger.info(f"Final user tasks: {user_tasks}")
|
|
|
|
save_progress(args['app_id'], current_step, {"user_tasks": user_tasks, "app_data": generate_app_data(args)})
|
|
|
|
return user_tasks
|
|
# USER TASKS END |