o i R@s^ddlmZmZddlmZddlmZddlmZddl m Z ddl m Z ddl mZddlZddlZddlZdd lmZed d gd Zed dejfddZeddejfddZdedefddZdedefddZdedefddZ dededefd d!Z!ded"ed#ed$ed%ed&edefd'd(Z"e#d)defd*d+Z$e#d,d-d.Z%dS)/) APIRouter HTTPException) FileResponse)studyplan_schema)studyplan_module)studyplan_questions)fill_excel_with_data)loggerN)Pathz/goal-skill-t/api/studyplan StudyPlan)prefixtagsz/startrequestc sz,t}|stddd|d}td|j|d|d|dd d t|d WStyI}zt d |tdt |dd }~ww)u 학습계획표 시작 - 첫 번째 질문 반환 Returns: - question_number: 질문 번호 (1) - question_text: 질문 내용 - current: 현재 질문 번호 (1) - total: 전체 질문 수 (5) zNo questions available status_codedetailrzStudyPlan Started for session: question_numbertextreasonr question_textrcurrenttotalzStudyPlan Start Error: N) r get_questionsrr info session_idgetlen Exceptionerrorstr)r all_questionsfirst_questioner'>/home/air/goalskill_sales/back/app/routers/studyplan_router.pystart_studyplans$   r)z/answerc szpt|j}|stdddtj|j|j|d|jd|j}tt }t d|d|d|j||krTt |jId H}d d |rOd |jd WSd d WSt }||}d|d|d| dd|d|ddWSty}zt d|tdt|dd }~ww)u 학습계획표 답변 제출 Returns: - status: "continue" (계속) or "completed" (완료) - next_question: 다음 질문 (continue인 경우) - result: 완료 메시지 (completed인 경우) zQuestion not foundrr)rrr answer_textzStudyPlan Progress: / for session N completedu_学習計画表の入力が完了しました！ 学習計画表のPDFを生成しました。z/studyplan/pdf/)statusmessagepdf_urlcontinuerrrrr)r/ next_questionzStudyPlan Answer Error: r)rget_question_by_numberrrrsave_studyplan_questionrr+r rr rprocess_final_resultrr!r"r#)rcurrent_questionanswered_counttotal_questionspdf_pathr$r3r&r'r'r(answer_studyplan9sJ     r;rreturnc szddl}t|}t|dkr2tdt|d|td|tddt|d d t|IdH}|d |dd  }|d |dd  }|dt |dd  }|dt |dd  }|d|dd  }t dt ddd|Dt d|d|t d|dt |dt d|dt |dt d|t d tj ||||||d!tj||||||d"t||||||d!IdH} t d#|d$| | WSty} z td%| | d} ~ ww)&uZ 모든 질문 완료 시 최종 결과 처리 - studyplan_log에서 답변들 가져오기 - 답변 파싱 (데이터베이스 스키마에 맞게) - studyplan_Output에 저장 - user_profile_summary에 동기화 - BASE 엑셀 복사 → 데이터 채우기 → PDF 생성 Returns: PDF 파일 경로 rNzExpected 5 answers, but got r-z Answers: iz Not all questions answered. Got z answers instead of 5.rtotal_durationr+current_statusr weekday_hours weekend_hourspreferred_time_slotz === StudyPlan Parsing Result ===zOriginal answers: cSsg|]}|dqS)r+r').0ar'r'r( sz(process_final_result..zParsed data: duration=z , status=zParsed weekday_hours=z (type: )zParsed weekend_hours=zParsed time_slot=z ================================)rr>r?r@rBrD)rstudyplan_durationstudyplan_statusstudyplan_weekday_hoursstudyplan_weekend_hoursstudyplan_time_slotz.StudyPlan Final Result Processed for session: z, PDF: zStudyPlan Final Result Error: )rerget_answers_by_sessionr r r"rparse_answers_with_airstripextract_numberrtypesave_studyplan_outputsync_to_user_profile_summarygenerate_studyplan_scheduler!) rrOanswers parsed_datar>r?r@rBrDr:r&r'r'r(r6wsh         r6rXc sz|ddlm}ddl}d}t|dD]\}}|d|d|dd |d|d d 7}qd |d }|}|jjd|dddd}|j} | drb| dd} | dr^| dd} | } | | } t | dd| d<t | dd| d<| WSt y} z9td| d|dd |dd t|dd t|dd |dd dWYd} ~ Sd} ~ ww)u AI를 사용하여 5개의 답변을 분석하고 구조화된 데이터 추출 Args: answers: 답변 리스트 (question_number, question_text, answer_text 포함) Returns: { 'total_duration': str, # 예: "6ヶ月" 'current_status': str, # 예: "会社員" 'weekday_hours': int, # 예: 1 'weekend_hours': int, # 예: 3 'preferred_time_slot': str # 예: "夕方" } r)get_gemini_modelNrrQz: rz Ar+z um 以下の5つの質問と回答を分析して、構造化されたデータを抽出してください。 u 以下のJSON形式で返答してください。数字は必ず整数で返してください。 { "total_duration": "目標期間（例: 6ヶ月）", "current_status": "現在の状態（例: 会社員）", "weekday_hours": 平日の学習時間（整数、例: 1）, "weekend_hours": 週末の学習時間（整数、例: 3）, "preferred_time_slot": "希望時間帯（例: 夕方、夜間など、元のテキストそのまま）" } 重要: - weekday_hours: Q3の回答から平日（月〜金）の学習時間を抽出してください。ユーザーが「1時間」と言ったら1、「2時間」と言ったら2です。 - weekend_hours: Q4の回答から週末の学習時間を抽出してください。ユーザーが「3時間」と言ったら3、「6時間」と言ったら6です。 - weekday_hoursとweekend_hoursは必ず整数で返してください。推測や補完をせず、ユーザーが実際に言った数字だけを返してください。 - preferred_time_slotはユーザーが入力した元のテキストをそのまま返してください（例: "夕方"、"夜間"など） - JSONのみを返答してください。他の説明は不要です。 zgemini-2.5-flashg?zapplication/json) temperatureresponse_mime_type)modelcontentsconfigz```jsonrEr@rBzAI parsing error: z#, falling back to simple extractionrArC)r>r?r@rBrD)app.core.configrZra enumeratemodelsgenerate_contentrrR startswithsplitloadsintrr!r r"rS) rXrZraqa_textianswerpromptclientresponse result_textparsedr&r'r'r(rQsL ,    rQrcCs(ddl}|d|}|rt|SdS)uQ 텍스트에서 숫자 추출 예: "2時間" → 2, "6時間" → 6 rNz\d+)rOsearchrigroup)rrOmatchr'r'r(rSs   rS excel_path output_dirc Cszntd|tdtj|ddddddd ||g }td d |tj|d d d d d}td|j|j rJt d|j tj tj |dd}tj||}tj|slt d||WSt y}zdt|vsdt|vrtdtd|d}~wtjy}ztd|jtd|j d}~wtjytdty}z td|d}~ww)u LibreOffice를 사용하여 Excel 파일을 PDF로 변환합니다. Args: excel_path: 엑셀 파일 경로 output_dir: PDF 출력 디렉토리 Returns: 생성된 PDF 파일 경로 zConverting Excel to PDF: zExcel file exists: libreofficez --headlessz --nodefaultz --nolockcheckz --convert-tozJpdf:calc_pdf_Export:{"SinglePageSheets":{"type":"boolean","value":"true"}}z--outdirzLibreOffice command:  T)capture_outputrchecktimeoutz#LibreOffice PDF conversion stdout: z#LibreOffice PDF conversion stderr: r.pdfzPDF file was not created: z not foundzZERROR: 'libreoffice' command not found. Is LibreOffice installed and in the system's PATH?zERROR: Nz3ERROR: LibreOffice conversion failed. Return code: zStderr: z'ERROR: LibreOffice conversion timed outz4An unexpected error occurred during PDF conversion: )r rospathexistsjoin subprocessrunstdoutstderrwarningsplitextbasenameFileNotFoundErrorr#lowerr"CalledProcessError returncodeTimeoutExpiredr!)rurvcommandresult pdf_filenamer:r&r'r'r(convert_excel_to_pdf+sV     rr>r?r@rBrDc sZzttjjj}|d}|jdd|d}|s-td|tdWdS|d|d } t || t t | |||||d t t | t |d } t d | t d tj| dt d| t dtj| dt d| d| WSty} ztd| ddl} t| WYd} ~ dSd} ~ ww)u 학습계획표 시간표 엑셀 파일 생성 및 PDF 변환 BASE 엑셀 템플릿을 복사해서 유저 데이터를 채운 후 LibreOffice로 PDF 변환 Args: session_id: 세션 ID total_duration: 목표 기간 current_status: 현재 상태 weekday_hours: 평일 학습 시간 weekend_hours: 주말 학습 시간 preferred_time_slot: 선호 시간대 Returns: PDF 파일 경로 studyplan_filesT)exist_okzbase_studyplan_template.xlsxu'BASE 엑셀 템플릿이 없습니다: uT먼저 'python create_base_excel.py'를 실행하여 BASE 엑셀을 생성하세요.N studyplan_z.xlsx)rur>r?r@rBrD)rurvzStudyPlan Excel generated: zStudyPlan Excel file size: z byteszStudyPlan PDF generated: zStudyPlan PDF file size: z+=== Excel file available for manual check: z ===z StudyPlan PDF Generation Error: r)r __file__resolveparentmkdirrr r"shutilcopy2rr#rrr~rgetsizer! traceback format_exc) rr>r?r@rBrDbase_dir studyplan_dirbase_excel_pathuser_excel_pathr:r&rr'r'r(rWisJ    rWz/pdf/{session_id}c s(zvddl}ttjjj}|dd|d}|s#tddddd lm}t |d  }| }Wdn1s=wYd |d}|j j |d d }ddl } tj|} dt| d} ||dd|d|dddd| dddWSty} ztd| tdt| dd} ~ ww)u7 학습계획표 PDF 다운로드/미리보기 rNrrr}r*zPDF file not foundrResponserbu学習計画表_rsafe"application/pdfzinline; filename="studyplan_z.pdf"; filename*=UTF-8''#no-cache, no-store, must-revalidateno-cache0*)Content-Disposition Content-Type Cache-ControlPragmaExpiresETagAccess-Control-Allow-Origincontent media_typeheaderszStudyPlan PDF Download Error: r) urllib.parser rrrrrfastapi.responsesropenreadparsequotetimer~rgetmtimerir!r r"r#) rurllibrr:rf pdf_content filename_utf8filename_encodedr file_mtimeetagr&r'r'r(get_studyplan_pdfsB      rz/admin-base-curriculumc s4zuttjjj}|dd}td||s+td|tdddtd|t |d  }| }Wd n1sGwYd d l }d}d }|j j |d d}d dlm}||dd|d|dddddddWSty~ty} ztd| tdt| dd } ~ ww)u{ 학습계획표 양식 PDF 파일 서빙 /home/air/goalskill_t/back/studyplan_files/admin_base_curriculum.pdf rzadmin_base_curriculum.pdfz(Admin Base Curriculum PDF Request: path=z%Admin Base Curriculum PDF not found: r*z#Admin base curriculum PDF not foundrz!Admin Base Curriculum PDF found: rNru&学習計画表_基本フォーム.pdfrrrrzinline; filename="z"; filename*=UTF-8''rrrr)rrrrrrrz!Admin Base Curriculum PDF Error: r)r rrrr rrr"rrrrrrrrr!r#) rr:rrrfilename_asciirrrr&r'r'r(get_admin_base_curriculumsF      r)&fastapirrrr app.schemasr app.modelsr app.servicesr&app.services.studyplan_excel_generatorrrbr r~rrpathlibr routerpostStartStudyPlanRequestr)AnswerStudyPlanRequestr;r#r6listdictrQrirSrrWrrrr'r'r'r(sH       #=QV > F0