Update app.py

app.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import os
 import time
 import google.generativeai as genai
@@ -46,7 +47,10 @@ generation_config = {
 
 # 사용할 모델 이름 설정 (변경 가능)
 # 예: "gemini-1.5-flash-latest", "gemini-1.5-pro-latest" 등
-model_name = "gemini-2.5-flash-preview-04-17" # 최신 flash 모델 사용 권장
+# gemini-2.5-flash-preview-04-17 모델 이름이 유효하지 않을 수 있습니다.
+# 최신 flash 모델인 "gemini-1.5-flash-latest" 또는 pro 모델 "gemini-1.5-pro-latest" 사용을 권장합니다.
+# model_name = "gemini-2.5-flash-preview-04-17" # -> 유효하지 않을 가능성 있음
+model_name = "gemini-2.5-flash-preview-04-17" # 안정적인 최신 flash 모델로 변경
 
 try:
     model = genai.GenerativeModel(
@@ -58,6 +62,7 @@ try:
     # st.info(f"'{model_name}' 모델 로드 성공!") # 디버깅용
 except Exception as e:
     st.error(f"Gemini 모델 ('{model_name}') 로딩 중 오류 발생: {e}")
+    st.error("올바른 모델 이름인지 확인하거나 Google AI Studio에서 사용 가능한 모델 목록을 확인하세요.")
     st.stop()
 
 # --- 함수 정의 ---
@@ -79,13 +84,13 @@ def generate_text_and_questions(grade, num_paragraphs, sentences_per_paragraph,
     - 초등학생이 이해하기 쉬운 단어와 문장 사용
     - 줄글 형식으로 작성, 문단 구분 명확히 (빈 줄 사용)
     - 각 문단의 첫 문장 또는 마지막 문장이 중심 문장이 되도록 작성
-- **제목:** 글의 맨 처음에 내용을 잘 나타내는 제목을 **크고 굵은 글씨**로 작성 (예: '**제목**')
+- **제목:** 글의 맨 처음에 내용을 잘 나타내는 제목을 **크고 굵은 글씨**로 작성 (Markdown 형식: '** 제목 **')
 
 ## 2. 어휘 목록 작성
 - **선정 기준:** 본문 내용 중 초등학교 {grade}에게 어려울 수 있는 단어, 한자어, 학습 용어
 - **설명 방식:** 해당 학년 수준에 맞는 쉬운 유의어 또는 풀이 제공
 - **형식:** 설명문 본문 뒤에 '### 어휘 목록' 제목 아래 글머리 기호 목록으로 작성
-    ```
+    ```markdown
     ### 어휘 목록
     *   단어1: 쉬운 설명/유의어
     *   단어2: 쉬운 설명/유의어
@@ -106,7 +111,7 @@ def generate_text_and_questions(grade, num_paragraphs, sentences_per_paragraph,
 - **형식:**
     - '### 독해 문제' 제목 아래 문제 번호(1., 2., ...)와 함께 문제 제시
     - 모든 문제 출제 후, '### 정답' 제목 아래 문제 번호와 정답 명확히 제시
-    ```
+    ```markdown
     ### 독해 문제
     1. [문제 내용 - 단답형 또는 5지 선다형]
        ① 선택지1 ② 선택지2 ③ 선택지3 ④ 선택지4 ⑤ 선택지5 (선다형의 경우)
@@ -121,53 +126,67 @@ def generate_text_and_questions(grade, num_paragraphs, sentences_per_paragraph,
 
 ## # 출력 요구사항
 - 위의 모든 지시사항(설명문, 어휘 목록, 독해 문제, 정답)을 **하나의 응답**으로 이어서 생성해주세요.
-- 각 섹션(제목, 본문, 어휘 목록, 독해 문제, 정답) 구분을 명확히 해주세요.
+- 각 섹션(제목, 본문, 어휘 목록, 독해 문제, 정답) 구분을 Markdown 제목(##, ###)과 빈 줄로 명확히 해주세요.
 - 모든 내용은 초등학교 {grade} 눈높이에 맞춰 작성해주세요.
 """
 
     full_response_text = "" # 전체 응답 저장 변수 초기화
     output_area = st.empty() # 스트리밍 출력을 위한 빈 공간 생성
+    blocked = False # 콘텐츠 차단 플래그
 
     try:
         response = model.generate_content(prompt, stream=True)
         for chunk in response:
-            # 응답 후보(candidates)가 있고, 그 안에 내용(content)이 있고, 그 안에 부분(parts)이 있고, 그 안에 텍스트(text)가 있는지 확인
-            if chunk.candidates and chunk.candidates[0].content and chunk.candidates[0].content.parts and chunk.candidates[0].content.parts[0].text:
-                chunk_text = chunk.candidates[0].content.parts[0].text
+            # 프롬프트 피드백 확인 (차단 여부)
+            if hasattr(chunk, 'prompt_feedback') and chunk.prompt_feedback.block_reason:
+                block_reason = chunk.prompt_feedback.block_reason
+                st.error(f"콘텐츠 생성 요청이 안전상의 이유로 차단되었습니다. 이유: {block_reason}")
+                output_area.error(f"⚠️ 콘텐츠 생성 중단됨 (이유: {block_reason}). 프롬프트나 주제를 수정하여 다시 시도해 보세요.")
+                blocked = True
+                break # 차단 시 스트리밍 중단
+
+            # 응답 텍스트 처리
+            try:
+                # 최신 API는 chunk.text로 바로 접근 가능할 수 있음
+                chunk_text = chunk.text
+            except AttributeError:
+                 # 이전 방식 호환 (candidates 확인)
+                if chunk.candidates and chunk.candidates[0].content and chunk.candidates[0].content.parts and chunk.candidates[0].content.parts[0].text:
+                    chunk_text = chunk.candidates[0].content.parts[0].text
+                else:
+                    # 예상치 못한 청크 형식일 경우 건너뛰기 또는 로깅
+                    # st.warning(f"Unexpected chunk format: {chunk}") # 디버깅용
+                    chunk_text = "" # 또는 continue
+
+            if chunk_text:
                 full_response_text += chunk_text
-                # Markdown을 HTML로 변환하여 스트리밍 출력 (표 확장 기능 포함)
-                # nl2br 확장 기능으로 마크다운 줄바꿈(\n)을 <br>로 변환
+                # Markdown을 HTML로 변환하여 스트리밍 출력 (표, 코드 블록, 줄바꿈 지원)
                 html_text = markdown.markdown(full_response_text, extensions=['markdown.extensions.tables', 'markdown.extensions.fenced_code', 'markdown.extensions.nl2br'])
                 output_area.markdown(html_text, unsafe_allow_html=True)
                 time.sleep(0.05) # 스트리밍 지연 (조절 가능)
-            elif hasattr(chunk, 'prompt_feedback') and chunk.prompt_feedback.block_reason:
-                 # 콘텐츠 차단 발생 시 사용자에게 알림
-                block_reason = chunk.prompt_feedback.block_reason
-                st.error(f"콘텐츠 생성 중 안전상의 이유로 차단되었습니다. 이유: {block_reason}")
-                output_area.error(f"⚠️ 콘텐츠 생성 중단됨 (이유: {block_reason}). 프롬프트나 주제를 수정하여 다시 시도해 보세요.")
-                return "" # 차단 시 빈 문자열 반환
 
     except Exception as e:
         st.error(f"텍스트 생성 중 에러 발생: {str(e)}")
-        # 오류 발생 시 이전 내용을 유지하기 위해 output_area를 그대로 둠
+        # 오류 발생 시 현재까지 생성된 내용을 유지하도록 output_area 수정하지 않음
         return "" # 오류 시 빈 문자열 반환
 
-    # 생성 완료 후 복사 버튼 표시
-    st.markdown("---") # 구분선 추가
-    if full_response_text: # 생성된 내용이 있을 때만 복사 버튼 표시
-        # full_response_text 를 직접 사용 (html_text는 렌더링용)
-        # st.code(full_response_text, language='markdown') # 디버깅용: 원본 마크다운 확인
-
+    # 스트리밍 완료 후 처리 (차단되지 않았을 경우)
+    if not blocked and full_response_text:
+        st.markdown("---") # 구분선 추가
         copy_button = st.button("📄 생성된 내용 전체 복사")
         if copy_button:
             try:
-                # 마크다운 형식으로 복사 (HTML 태그 제외)
+                # 마크다운 형식으로 복사
                 pyperclip.copy(full_response_text)
                 st.success("클립보드에 복사되었습니다! (마크다운 형식)")
             except Exception as e:
-                st.warning(f"클립보드 자동 복사 중 오류 발생: {e}\n아래 텍스트 영역에서 수동으로 복사해주세요.")
+                # pyperclip 에러 처리 (예: 서버 환경 등에서 클립보드 접근 불가 시)
+                st.warning(f"클립보드 자동 복사 중 오류 발생: {e}\n서버 환경에서는 클립보드 복사가 지원되지 않을 수 있습니다.\n아래 텍스트 영역에서 수동으로 복사해주세요.")
                 # 복사 실패 시 수동 복사를 위해 텍스트 영역 추가 표시
-                st.text_area("복사할 내용 (마크다운):", full_response_text, height=200)
+                st.text_area("복사할 내용 (마크다운):", full_response_text, height=300)
+
+    elif blocked:
+        return "" # 차단 시 빈 문자열 반환
 
     return full_response_text # 전체 생성된 텍스트 반환 (마크다운 원본)
 
@@ -190,7 +209,9 @@ with st.sidebar:
 
     # 학년 선택 (초1 ~ 초6)
     grade_options = [f"{i}학년" for i in range(1, 7)]
-    grade = st.selectbox("대상 학년", grade_options, index=2) # 기본값 3학년
+    # 문자열에서 숫자만 추출하도록 수정
+    grade_str = st.selectbox("대상 학년", grade_options, index=2) # 기본값 3학년
+    grade = grade_str.replace("학년", "") # '3학년' -> '3'
 
     # 문단 수 설정
     num_paragraphs = st.number_input("문단 수 (권장 3~5)", min_value=2, max_value=10, value=3)
@@ -234,19 +255,22 @@ if generate_button:
         with st.spinner("AI가 열심히 글을 쓰고 문제를 만들고 있어요... 잠시만 기다려주세요! 🤔✍️"):
             # 함수 호출 (출력은 함수 내부에서 스트리밍으로 처리됨)
             generated_content = generate_text_and_questions(
-                grade,
+                grade, # 숫자 학년 전달
                 num_paragraphs,
                 sentences_per_paragraph,
                 structure,
                 topic
             )
-            # 생성 완료 후 별도 메시지 (필요시, 함수 내 복사 버튼 표시로 대체 가능)
+            # 생성 완료 후 별도 메시지는 함수 내에서 처리 (복사 버튼 표�� 등)
+            # 또는 스피너 완료 자체가 성공 표시가 됨
+            # 주석 처리된 부분의 pass 삭제: IndentationError 수정
             # if generated_content:
-            #     st.success("읽기 자료 생성이 완료되었습니다!")
+            #     st.success("읽기 자료 생성이 완료되었습니다!") # 성공 메시지 필요 시 주석 해제
             # elif not generated_content: # 생성된 내용이 없는 경우 (오류 또는 차단)
                  # 오류 메시지는 함수 내부에서 이미 표시됨
                  # st.error("읽기 자료 생성에 실패했습니다.")
-                 pass # 함수 내에서 이미 에러 또는 차단 메시지 표시
+                 # pass # <- 이 줄이 IndentationError의 원인. 삭제됨.
+                 pass # 이 pass는 else 블록에 속하며, 현재는 아무 작업도 하지 않음을 나타냄 (필요시 로직 추가 가능)
 
 # 만약 스트리밍 출력을 함수 밖에서 제어하고 싶다면,
 # output_area = st.empty() 를 여기 두고, 함수는 text chunk 만 반환하게 수정 필요