#YhddlZddlZddlmZddlZddlZddlZeddZdZdZ dZ d Z d Z d Z d Zd ZdZdZy)N)pipelinezquestion-answeringz'distilbert-base-uncased-distilled-squad)modelc$tjddddd}|jtjj}|j d||f|j }|jtjd||S) N localhost openai_user,IOyg76H2l%252BewRX2xhsDJAo7qnfVDHtx9RB%253D%openai_promptsutf8mb4hostuserpasswddbcharsetz SELECT instruction_text, question_text, ai_prompt_id, feedbackRequest_text, prompt_template_name, section_label FROM phase_content WHERE app_id = %s AND phase_number = %s u$🔎 QA_SYSTEM: Phase data fetched: ) MySQLdbconnectcursorcursors DictCursorexecutefetchonecloselogginginfo)app_idphaseconnrresults (/var/www/html/decompression/qa_system.py get_next_partr s ??  =   D[[33 4F NN %  __ FJJL LL7x@A Mctjddddd}|jtjj}|j d|f|j }|s|jgS|d}|j d |f|j}|j|Dcgc] }t|c}Scc}w) Nrrrr r r zE SELECT id FROM prompt_templates WHERE name = %s idzn SELECT role, content FROM prompt_messages WHERE prompt_id = %s ORDER BY sequence ) rrrrrrrrfetchalldict)namerrrow prompt_idmessagesmsgs r$get_prompt_messages_by_template_namer+s ??  =   D[[33 4F NN  // C   D I NN    HJJL!) *#DI ** *s7C cLtjdd|jS)z Removes headers that are all uppercase and surrounded by dashes. :param text: The text to process. :return: Text with headers removed. z -[A-Z\s]+-)resubstriptexts rremove_headersr3As  66-T * 0 0 22r!c&t||}|dS)z Get an answer from the model based on a question and context. :param question: The question asked :param context: The context for the question :return: The model's answer )questioncontextanswer) qa_pipeline)r5r6rs r get_answerr9Ks(G  T^_``+L(99)DJJLN  3N4C4H3I MN r!cB|jddjS)z Sanitizes the text by removing unwanted characters. :param text: The text to sanitize. :return: Sanitized text.  r-)replacer0r1s r sanitize_textrQs <<b ! ' ' ))r!cddddddgS)z{ Returns default questions if no specific questions are found for a phase. :return: List of default questions. z(A) Proceed to the next phase.A)r2choicez(B) Repeat the current phase.BrVr!rget_default_questionsrWs 2SA0C@ r!cHddddddddd d d d d dd}|j|dS)z Returns the context for a given phase, used as input for the AI model. :param phase: The current phase :return: Context string for the phase zContext for phase 1zContext for phase 2zContext for phase 3zContext for phase 4zContext for phase 5zContext for phase 6zContext for phase 7zContext for phase 8zContext for phase 9zContext for phase 10zContext for phase 11zContext for phase 12zContext for phase 13zContext for phase 14) zUnknown phase context)get)rcontext_mappings rget_phase_contextrisK ! " " " " "O   u&= >>r!cv |jdd}ddddj|dS#t$rYywxYw)N:rBrYrZr[)500050015002)splitrg Exception)r ports rdetect_app_id_from_portrrsMzz#r"  #dA,   s ), 88)jsonr. transformersrrosrr8r r+r3r9r@rMrQrWrirrrVr!rrvs[ ! +3\] (+D3 W8*?0 r!