Create app.py

app.py

@@ -0,0 +1,137 @@
+import gradio as gr
+import google.generativeai as genai
+from transformers import pipeline
+import os
+from typing import IO # 파일 타입 힌트를 위해 추가
+
+# Hugging Face Spaces의 메모리 제한을 고려하여 경량 모델을 사용합니다.
+# 한국어 요약에 특화된 모델을 로드합니다.
+try:
+    summarizer = pipeline("summarization", model="gogamza/kobart-summarization")
+except Exception as e:
+    print(f"모델 로딩 중 오류 발생: {e}")
+    summarizer = None
+
+# Gradio 4.x에 맞춰 함수 시그니처를 수정합니다.
+# progress 인자를 타입 힌트로 받습니다.
+def generate_blog_post(api_key: str, file1: IO, file2: IO, file3: IO, progress: gr.Progress):
+    """
+    3개의 텍스트 파일을 입력받아 요약하고, Gemini API를 통해 블로그 글을 생성하는 함수
+    """
+    # 1. 입력 값 유효성 검사
+    if not api_key:
+        raise gr.Error("Google API 키를 입력해 주세요.")
+    if not (file1 and file2 and file3):
+        raise gr.Error("3개의 텍스트 파일을 모두 업로드해 주세요.")
+    if not summarizer:
+        raise gr.Error("요약 모델을 로드하는 데 실패했습니다. 앱 로그를 확인해 주세요.")
+
+    try:
+        # 2. Google Gemini API 설정
+        progress(0.1, desc="Google API 키 설정 중...")
+        genai.configure(api_key=api_key)
+
+        # 3. 텍스트 파일 읽기
+        progress(0.2, desc="텍스트 파일 읽는 중...")
+        texts = []
+        # Gradio 4.x에서 File 객체는 임시 파일 경로를 name 속성으로 가집니다.
+        files = [file1, file2, file3]
+        for f in files:
+            with open(f.name, 'r', encoding='utf-8') as file:
+                texts.append(file.read())
+
+        # 4. 허깅페이스 모델로 각 텍스트 요약
+        summaries = []
+        for i, text in enumerate(texts):
+            progress(0.3 + i * 0.1, desc=f"문서 {i+1} 요약 중...")
+            summary = summarizer(text, max_length=200, min_length=50, do_sample=False)
+            summaries.append(summary[0]['summary_text'])
+        
+        # 5. Gemini API로 새로운 글 생성
+        progress(0.6, desc="Gemini API로 블로그 아웃라인 생성 중...")
+        model = genai.GenerativeModel('gemini-1.5-flash-latest')
+
+        summaries_str = "\n".join([f"- 요약 {i+1}: {s}" for i, s in enumerate(summaries)])
+        
+        prompt = f"""
+        당신은 IT 전문 네이버 블로거입니다. 아래 3개의 요약문을 참고하여, 독자들의 흥미를 끌 수 있는 블로그 글을 작성해 주세요.
+
+        **작성 조건:**
+        1.  먼저 글의 전체적인 '아웃라인'을 서론, 본론, 결론 형식으로 명확하게 제시해 주세요.
+        2.  그 다음, 위에서 제시한 아웃라인에 맞춰서 전체 글을 작성해 주세요.
+        3.  네이버 블로그 스타일에 맞게 친근하고 이해하기 쉬운 어투를 사용하고, 적절히 이모지를 사용해 주세요.
+        4.  SEO를 고려하여 '인공지능', 'AI 기술', '미래 사회', '허깅페이스', 'Gemini API'와 같은 키워드를 본문에 자연스럽게 포함시켜 주세요.
+        5.  전체 글자 수는 공백 포함 3,000자 내외로 맞춰주세요.
+        6.  결과는 마크다운 형식으로 깔끔하게 정리해서 보여주세요.
+
+        **[참고할 요약문]**
+        {summaries_str}
+
+        **[결과 출력 형식]**
+        # [블로그 제목]
+        
+        ## 📝 아웃라인
+        - **서론**: ...
+        - **본론**: ...
+        - **결론**: ...
+
+        ---
+
+        ## ✍️ 완성된 블로그 글
+        (여기에 완성된 글을 출력)
+        """
+
+        progress(0.8, desc="Gemini API로 최종 블로그 글 생성 중...")
+        response = model.generate_content(prompt)
+        
+        progress(1.0, desc="완료!")
+        return response.text
+
+    except Exception as e:
+        raise gr.Error(f"오류가 발생했습니다: {e}")
+
+
+# Gradio 인터페이스 구성
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown(
+        """
+        # 🤖 AI 기반 네이버 블로그 포스팅 자동 생성
+        3개의 텍스트 파일을 업로드하고 Google API 키를 입력하면, AI가 파일 내용을 분석하고 요약하여 네이버 블로그 형식의 새로운 글을 만들어 드립니다.
+        """
+    )
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            api_key = gr.Textbox(
+                label="Google API 키",
+                placeholder="여기에 Google API 키를 입력하세요.",
+                type="password",
+                lines=1
+            )
+            file1 = gr.File(label="텍스트 파일 1")
+            file2 = gr.File(label="텍스트 파일 2")
+            file3 = gr.File(label="텍스트 파일 3")
+            generate_btn = gr.Button("🚀 블로그 글 생성하기", variant="primary")
+
+        with gr.Column(scale=2):
+            output_text = gr.Markdown(label="✨ 생성된 블로그 글")
+
+    generate_btn.click(
+        fn=generate_blog_post, 
+        inputs=[api_key, file1, file2, file3], 
+        outputs=output_text
+    )
+    
+    gr.Markdown(
+        """
+        ---
+        **사용 방법:**
+        1.  [Google AI Studio](https://aistudio.google.com/app/apikey)에서 API 키를 발급받아 입력합니다.
+        2.  주제가 유사한 3개의 텍스트 파일(.txt)을 각각 업로드합니다.
+        3.  '블로그 글 생성하기' 버튼을 클릭하고 잠시 기다립니다.
+        4.  오른쪽 창에 생성된 글의 아웃라인과 최종 결과물이 나타납니다.
+        """
+    )
+
+if __name__ == "__main__":
+    demo.launch()