Update app.py

app.py

@@ -1,107 +1,132 @@
 import os
 import time
-import google.generativeai as genai
-from PyPDF2 import PdfReader
 import streamlit as st
+from PIL import Image  # 이미지 관련 라이브러리는 필요 없을 수 있지만, 일단 포함
+import google.generativeai as genai
+from streamlit_extras.colored_header import colored_header
 import markdown
 import pyperclip
+import PyPDF2  # PDF 처리 라이브러리 추가
 
-# Google Gemini API 설정 (Streamlit secrets 사용 권장)
+# Google Gemini API 키 설정 (기존 코드와 동일)
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
-# Gemini 모델 설정
+# 모델 설정 (기존 코드와 동일)
 generation_config = {
-    "temperature": 1,
+    "temperature": 0.7, # temperature 값 조정 (보고서/계획서는 창의성보다 정확성이 중요할 수 있음)
     "top_p": 0.95,
     "top_k": 40,
-    "max_output_tokens": 8192,
+    "max_output_tokens": 4096,
     "response_mime_type": "text/plain",
 }
 
-def extract_text_from_pdf(pdf_file):
+model = genai.GenerativeModel(
+    model_name="gemini-2.0-flash-exp", # 필요에 따라 모델 변경 가능 (pro, flash 등)
+    generation_config=generation_config,
+)
+
+# PDF 텍스트 추출 함수
+def extract_text_from_pdf(uploaded_pdf_file):
     """PDF 파일에서 텍스트를 추출합니다."""
+    text = ""
     try:
-        pdf_reader = PdfReader(pdf_file)
-        text = ""
+        pdf_reader = PyPDF2.PdfReader(uploaded_pdf_file)
         for page_num in range(len(pdf_reader.pages)):
             page = pdf_reader.pages[page_num]
-            text += page.extract_text() or ""
-        return text
+            text += page.extract_text()
     except Exception as e:
-        raise Exception(f"PDF 파일을 읽는 동안 오류가 발생했습니다: {e}")
-
-def generate_report(report_format_pdf, reference_pdfs, additional_instructions: str = "") -> str:
-    """Gemini를 사용하여 보고서를 생성합니다."""
+        st.error(f"PDF 파일 처리 오류: {e}")
+        return None
+    return text
+
+def generate_report_or_plan(pdf_text, user_instructions):
+    """
+    PDF 텍스트와 사용자 지시사항을 기반으로 보고서 또는 계획서를 생성합니다.
+    Args:
+        pdf_text: 추출된 PDF 텍스트
+        user_instructions: 사용자 지시사항
+    Returns:
+        str: 생성된 보고서 또는 계획서 텍스트
+    """
+    full_text = ""
     try:
-        # 보고서 서식 PDF에서 텍스트 추출
-        report_format = extract_text_from_pdf(report_format_pdf)
-
-        # 참고 PDF 파일에서 텍스트 추출
-        reference_text = ""
-        for pdf_file in reference_pdfs:
-            reference_text += extract_text_from_pdf(pdf_file)
-
-        # 프롬프트 생성
-        prompt = (
-            f"다음 서식을 사용하여 보고서를 작성하십시오:\n\n"
-            f"{report_format}\n\n"
-            f"다음 참고 자료를 사용하십시오:\n\n"
-            f"{reference_text}\n\n"
-            f"추가 지침:\n\n"
-            f"{additional_instructions}\n\n"
-            f"보고서를 최대한 자세하고 유익하게 작성하십시오."
-        )
-
-        # Gemini API 호출
-        response = genai.generate_text(prompt, **generation_config)
-        if hasattr(response, 'result'):
-            full_text = response.result
-        elif isinstance(response, dict) and 'result' in response:
-            full_text = response['result']
-        else:
-            raise Exception("Gemini API로부터 올바르지 않은 응답을 받지 못했습니다.")
+        prompt_text = f"""
+        # PDF 문서 분석 및 보고서/계획서 작성
+
+        ## PDF 문서 내용:
+        ```
+        {pdf_text}
+        ```
 
-        return full_text
+        ## 사용자 지시사항:
+        {user_instructions}
+
+        ---
+
+        **지시사항에 따라 PDF 문서 내용을 분석하고, 보고서 또는 계획서를 작성하세요.**
+        **한국어로 작성하며, 명확하고 논리적인 구조로 작성해주세요.**
+        **필요하다면, PDF 문서 내용을 참고하여 추가적인 정보를 생성하거나 확장해도 좋습니다.**
+        **만약 PDF 내용이 부족하거나 지시사항을 수행하기 어렵다면, 솔직하게 답변해주세요.**
+
+        """ # 프롬프트 개선: 더 명확하고 구체적인 지시 포함
+
+        response = model.generate_content([prompt_text]) # 이미지 대신 텍스트 프롬프트만 사용
+        for chunk in response.text:
+            full_text += chunk
+            time.sleep(0.01)
+            st.session_state.generated_result += chunk
+            output_area.markdown(st.session_state.generated_result, unsafe_allow_html=True)
     except Exception as e:
-        st.error(f"보고서 생성 중 오류가 발생했습니다: {e}")
-        raise
-
-# Streamlit 앱 설정
-st.title("보고서 생성기")
-st.write("서식 PDF, 참고 PDF 파일, 추가 지침을 입력하여 보고서를 생성하세요.")
-
-# 파일 업로드
-report_format_pdf = st.file_uploader("보고서 서식 PDF 업로드", type="pdf")
-reference_pdfs = st.file_uploader("참고 PDF 파일 업로드", type="pdf", accept_multiple_files=True)
-additional_instructions = st.text_area("추가 지침", height=100)
-
-# 출력 영역을 함수 외부에 정의
-output_area = st.empty()
-
-# 보고서 생성 버튼
-if st.button("보고서 생성") and report_format_pdf and reference_pdfs:
-    with st.spinner("보고서 생성 중..."):
-        try:
-            generated_report = generate_report(report_format_pdf, reference_pdfs, additional_instructions)
-            st.session_state.generated_report = generated_report
-
-            # 생성된 보고서를 출력
-            html_text = markdown.markdown(generated_report, extensions=['tables'])
+        st.error(f"보고서/계획서 생성 중 오류 발생: {str(e)}")
+    return full_text
+
+# Streamlit 인터페이스 설정 (기존 코드와 유사)
+colored_header(
+    label="PDF 보고서/계획서 자동 작성 AI",
+    description="PDF 서식 파일을 업로드하고 지시사항을 입력하면 보고서/계획서를 작성해줍니다.",
+    color_name="blue-70", # 색상 변경
+)
+
+# 파일 업로드 버튼 (PDF 파일만 허용)
+uploaded_pdf_file = st.file_uploader("PDF 서식 파일 업로드", type=["pdf"])
+
+# 지시사항 입력 텍스트 영역 추가
+user_instructions = st.text_area("작성 지시사항 입력 (예: '이 PDF를 바탕으로 2024년 마케팅 계획 보고서를 작성해주세요.', '핵심 내용을 요약하고, 개선 방안을 3가지 제시해주세요.')",
+                                 placeholder="자세하고 명확하게 지시사항을 입력하세요.") # placeholder 추가
+
+generate_button = st.button("보고서/계획서 생성")
+
+output_area = st.empty() # 출력 영역 (기존 코드와 동일)
+
+if generate_button and uploaded_pdf_file and user_instructions: # 지시사항도 입력되었는지 확인
+    try:
+        pdf_text = extract_text_from_pdf(uploaded_pdf_file) # PDF 텍스트 추출
+        if pdf_text: # 텍스트 추출 성공 시
+            st.session_state.generated_result = "" # 결과 초기화
+            result = generate_report_or_plan(pdf_text, user_instructions) # 보고서/계획서 생성
+
+            html_text = markdown.markdown(result, extensions=['tables']) # Markdown 형식으로 변환 (필요에 따라 조정)
             output_area.markdown(html_text, unsafe_allow_html=True)
 
-            # 복사 버튼 추가
-            if st.button("출력 내용 복사", key="copy_button_1"):
-                pyperclip.copy(generated_report)
+            if st.button("출력 내용 복사", key="copy_button_1"): # 복사 버튼 (기존 코드와 동일)
+                pyperclip.copy(result)
                 st.success("복사되었습니다!")
-        except Exception as e:
-            st.error(e)
+        else:
+            st.warning("PDF 파일에서 텍스트를 추출하는데 실패했습니다. 파일 형식을 확인해주세요.")
+
+    except Exception as e:
+        st.error(f"보고서/계획서 생성 중 오류 발생: {str(e)}")
+
+elif generate_button and not uploaded_pdf_file:
+    st.warning("PDF 서식 파일을 업로드하세요.")
+elif generate_button and not user_instructions:
+    st.warning("작성 지시사항을 입력하세요.")
 
-# 이미 생성된 보고서가 있을 경우 이를 출력
-if "generated_report" in st.session_state and st.session_state.generated_report:
-    html_text = markdown.markdown(st.session_state.generated_report, extensions=['tables'])
+# 이미 생성된 결과가 있을 경우 출력 (기존 코드와 동일)
+if "generated_result" in st.session_state and st.session_state.generated_result:
+    html_text = markdown.markdown(st.session_state.generated_result, extensions=['tables'])
     output_area.markdown(html_text, unsafe_allow_html=True)
 
-    # 복사 버튼 추가 (이미 생성된 경우)
     if st.button("출력 내용 복사", key="copy_button_2"):
-        pyperclip.copy(st.session_state.generated_report)
+        pyperclip.copy(st.session_state.generated_result)
         st.success("복사되었습니다!")