Initial commit

.gitignore

@@ -0,0 +1,16 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+venv/
+.env
+
+# IDE
+.vscode/
+.idea/
+
+# Logs
+*.log
+
+# Environment variables
+.env 

README.md

@@ -0,0 +1,61 @@
+---
+title: My Diary AI
+emoji: 📝
+colorFrom: blue
+colorTo: pink
+sdk: gradio
+sdk_version: 4.8.0
+app_file: app.py
+pinned: false
+---
+
+AI를 활용한 일기 생성 프로젝트입니다.
+
+## 특징
+- 키워드를 기반으로 자세한 일기 생성
+- 감정과 상황을 생생하게 표현
+- Hugging Face의 한국어 언어 모델 활용
+
+## 설치 방법
+
+```bash
+git clone https://huggingface.co/Ping9gu/my-diary
+cd my-diary
+```
+
+## 저장소 클론
+```bash
+git clone https://github.com/Ping9gu/my-diary.git
+```
+
+## 가상환경 생성 및 활성화
+```bash
+python -m venv venv
+source venv/bin/activate
+```
+
+## 의존성 설치
+```bash
+pip install -r requirements.txt
+```
+
+## 환경 설정
+1. .env 파일 생성하고 다음 내용 추가:
+``` 
+HUGGINGFACE_API_KEY=your_token_here
+MODEL_ID=nlpai-lab/kullm-polyglot-5.8b-v2
+```
+
+## 서버 실행
+```bash
+cd backend
+./run_server.sh
+```
+
+## 사용 방법
+1. 웹 브라우저에서 http://localhost:5000 접속
+2. 키워드 입력
+3. "일기 생성하기" 버튼 클릭
+
+## 라이선스
+MIT License

app.py

@@ -0,0 +1,7 @@
+import gradio as gr
+from backend.app import create_interface
+
+demo = create_interface()
+
+if __name__ == "__main__":
+    demo.launch() 

backend/app.py

@@ -0,0 +1,128 @@
+from flask import Flask, request, jsonify, send_from_directory
+from flask_cors import CORS
+from huggingface_hub import InferenceClient
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+app = Flask(__name__)
+CORS(app, resources={
+    r"/*": {
+        "origins": ["http://your-frontend-domain"],
+        "methods": ["GET", "POST", "OPTIONS"],
+        "allow_headers": ["Content-Type", "Accept"]
+    }
+})
+
+# Hugging Face 클라이언트 초기화
+client = InferenceClient(
+    model=os.getenv("MODEL_ID"),
+    token=os.getenv("HUGGINGFACE_API_KEY")
+)
+
+# 더 큰 한국어 모델로 변경
+model_name = "nlpai-lab/kullm-polyglot-5.8b-v2"  # 더 나은 품질의 모델
+
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(
+    model_name,
+    device_map="auto",
+    torch_dtype=torch.float16,
+    low_cpu_mem_usage=True
+)
+
+# device_map="auto"를 사용하므로 .to(device) 호출 제거
+# device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+# model = model.to(device)  # 이 줄 제거
+
+# 루트 경로 핸들러 추가
+@app.route('/')
+def serve_frontend():
+    return send_from_directory('../frontend', 'index.html')
+
+# 정적 파일 서빙을 위한 라우트 추가
+@app.route('/<path:path>')
+def serve_static(path):
+    return send_from_directory('../frontend', path)
+
+# favicon.ico 핸들러 추가
+@app.route('/favicon.ico')
+def favicon():
+    return '', 204  # No Content 응답 반환
+
+@app.route('/api/generate-diary', methods=['POST'])
+def generate_diary():
+    try:
+        data = request.json
+        if not data or 'keywords' not in data:
+            return jsonify({"error": "키워드가 필요합니다"}), 400
+            
+        keywords = data.get('keywords', '').strip()
+        if not keywords:
+            return jsonify({"error": "키워드가 비어있습니다"}), 400
+        
+        prompt = f"""다음은 오늘 있었던 일의 요약입니다. 이것을 바탕으로 생생하고 감동적인 일기를 작성해주세요.
+
+        [상세 요구사항]
+        1. 도입부: 
+           - 그날의 날씨나 분위기로 시작
+           - 상황과 등장인물 소개
+        
+        2. 전개:
+           - 구체적인 대화와 행동 묘사
+           - 오감을 사용한 장면 묘사
+           - 등장인물들의 표정과 감정 변화
+        
+        3. 감정과 생각:
+           - 내면의 감정을 섬세하게 표현
+           - 사건에 대한 나의 생각과 깨달음
+           - 다른 사람들의 감정에 대한 공감
+        
+        4. 문체:
+           - 문어체와 구어체를 적절히 혼용
+           - 비유와 은유를 활용한 표현
+           - 반복을 피하고 다양한 어휘 사용
+        
+        5. 마무리:
+           - 그날의 경험이 주는 의미
+           - 앞으로의 기대나 다짐
+        
+        요약:
+        {keywords}
+        
+        ===
+        오늘의 일기:
+        오늘은 """
+        
+        # Hugging Face API를 통한 텍스트 생성
+        parameters = {
+            "max_new_tokens": 768,
+            "temperature": 0.88,
+            "top_p": 0.95,
+            "repetition_penalty": 1.35,
+            "top_k": 50,
+            "do_sample": True,
+            "num_return_sequences": 1
+        }
+        
+        response = client.text_generation(
+            prompt,
+            **parameters
+        )
+        
+        if not response:
+            return jsonify({"error": "일기 생성에 실패했습니다"}), 500
+            
+        # 프롬프트 제거하고 생성된 텍스트만 반환
+        diary_content = response.split("오늘은 ")[-1].strip()
+        diary_content = "오늘은 " + diary_content
+            
+        return jsonify({"diary": diary_content})
+    
+    except Exception as e:
+        print(f"Error generating diary: {str(e)}")
+        return jsonify({"error": f"일기 생성 중 오류가 발생했습니다: {str(e)}"}), 500
+
+if __name__ == '__main__':
+    app.run(debug=True)

backend/config.py

@@ -0,0 +1,8 @@
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+class Config:
+    SECRET_KEY = os.getenv('SECRET_KEY', 'your-secret-key')
+    DEEPSEEK_API_KEY = os.getenv('DEEPSEEK_API_KEY')

backend/models/diary.py

@@ -0,0 +1,14 @@
+from datetime import datetime
+
+class Diary:
+    def __init__(self, content, keywords):
+        self.content = content
+        self.keywords = keywords
+        self.created_at = datetime.now()
+        
+    def to_dict(self):
+        return {
+            'content': self.content,
+            'keywords': self.keywords,
+            'created_at': self.created_at.isoformat()
+        }

backend/requirements.txt

@@ -0,0 +1,7 @@
+flask==2.3.3
+flask-cors==4.0.0
+requests==2.31.0
+python-dotenv==1.0.0
+werkzeug==2.3.7
+bitsandbytes>=0.41.1
+huggingface-hub>=0.17.0

backend/run_server.sh

@@ -0,0 +1,5 @@
+#!/bin/bash
+screen -S diary-server -dm bash -c '
+source venv/bin/activate
+python app.py
+' 

frontend/index.html

@@ -0,0 +1,26 @@
+<!DOCTYPE html>
+<html lang="ko">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>AI 일기 도우미</title>
+    <link rel="stylesheet" href="style.css">
+    <link rel="icon" type="image/x-icon" href="/favicon.ico">
+</head>
+<body>
+    <h1>AI 일기 도우미</h1>
+    <form>
+        <textarea placeholder="오늘은 무슨 일이 있었어?"></textarea>
+        <button type="submit">일기 생성하기</button>
+    </form>
+    <div id="loading-container" style="display: none;">
+        <div class="loading-spinner"></div>
+        <div class="loading-text">일기를 생성하는 중입니다...</div>
+        <div class="loading-progress">
+            <div class="progress-bar"></div>
+        </div>
+    </div>
+    <div id="result"></div>
+    <script src="script.js"></script>
+</body>
+</html>

frontend/script.js

@@ -0,0 +1,60 @@
+document.addEventListener('DOMContentLoaded', function() {
+    const diaryForm = document.querySelector('form');
+    const keywordsInput = document.querySelector('textarea');
+    const resultDiv = document.querySelector('#result');
+    const generateButton = document.querySelector('button');
+    const loadingContainer = document.querySelector('#loading-container');
+    const progressBar = document.querySelector('.progress-bar');
+
+    diaryForm.addEventListener('submit', async function(e) {
+        e.preventDefault();
+        
+        const keywords = keywordsInput.value.trim();
+        if (!keywords) {
+            resultDiv.textContent = "키워드를 입력해주세요";
+            return;
+        }
+
+        try {
+            generateButton.disabled = true;
+            resultDiv.textContent = "";
+            loadingContainer.style.display = 'block';
+            progressBar.style.width = '0%';
+            
+            // 프로그레스 바 애니메이션 시작 (60초에서 30초로 변경)
+            progressBar.style.animation = 'progress 30s linear';
+            
+            const API_URL = 'http://your-ec2-instance-ip:5000';
+
+            const response = await fetch(`${API_URL}/api/generate-diary`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    'Accept': 'application/json'
+                },
+                body: JSON.stringify({
+                    keywords: keywords
+                })
+            });
+
+            if (!response.ok) {
+                const errorText = await response.text();
+                throw new Error(`서버 응답 오류: ${response.status}`);
+            }
+
+            const data = await response.json();
+            
+            if (data.error) {
+                resultDiv.textContent = `오류: ${data.error}`;
+            } else {
+                resultDiv.textContent = data.diary || '일기 생성에 실패했습니다.';
+            }
+        } catch (error) {
+            resultDiv.textContent = `오류가 발생했습니다: ${error.message}`;
+        } finally {
+            generateButton.disabled = false;
+            loadingContainer.style.display = 'none';
+            progressBar.style.animation = 'none';
+        }
+    });
+});

frontend/style.css

@@ -0,0 +1,95 @@
+body {
+    font-family: 'Noto Sans KR', sans-serif;
+    margin: 0;
+    padding: 20px;
+    background-color: #f5f5f5;
+}
+
+.container {
+    max-width: 800px;
+    margin: 0 auto;
+    background-color: white;
+    padding: 20px;
+    border-radius: 10px;
+    box-shadow: 0 0 10px rgba(0,0,0,0.1);
+}
+
+.input-section {
+    margin: 20px 0;
+}
+
+textarea {
+    width: 100%;
+    height: 100px;
+    padding: 10px;
+    margin-bottom: 10px;
+    border: 1px solid #ddd;
+    border-radius: 5px;
+}
+
+button {
+    background-color: #007bff;
+    color: white;
+    border: none;
+    padding: 10px 20px;
+    border-radius: 5px;
+    cursor: pointer;
+}
+
+button:hover {
+    background-color: #0056b3;
+}
+
+.diary-section {
+    margin-top: 20px;
+    padding: 20px;
+    border: 1px solid #ddd;
+    border-radius: 5px;
+}
+
+#loading-container {
+    text-align: center;
+    margin: 20px 0;
+}
+
+.loading-spinner {
+    width: 50px;
+    height: 50px;
+    border: 5px solid #f3f3f3;
+    border-top: 5px solid #3498db;
+    border-radius: 50%;
+    animation: spin 1s linear infinite;
+    margin: 0 auto;
+}
+
+.loading-text {
+    margin: 10px 0;
+    color: #666;
+}
+
+.loading-progress {
+    width: 100%;
+    max-width: 300px;
+    height: 4px;
+    background-color: #f3f3f3;
+    margin: 10px auto;
+    border-radius: 2px;
+    overflow: hidden;
+}
+
+.progress-bar {
+    width: 0%;
+    height: 100%;
+    background-color: #3498db;
+    animation: progress 60s linear;
+}
+
+@keyframes spin {
+    0% { transform: rotate(0deg); }
+    100% { transform: rotate(360deg); }
+}
+
+@keyframes progress {
+    0% { width: 0%; }
+    100% { width: 100%; }
+}