Upload 4 files

app.py

@@ -1,38 +1,158 @@
-from transformers import pipeline
 import gradio as gr
+import random
+import requests
+from PIL import Image, ImageDraw, ImageFont
+import io
+import base64
+from transformers import pipeline
+from config import MEME_TEMPLATES, CHATBOT_RESPONSES, APP_CONFIG
+from utils import create_meme_image, generate_response, add_text_to_image
+
+# AI 모델 초기화
+try:
+    qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distilled-squad")
+    text_generator = pipeline("text-generation", model="gpt2", max_length=100)
+except Exception as e:
+    print(f"모델 로딩 오류: {e}")
+    qa_pipeline = None
+    text_generator = None
+
+def create_meme(template_choice, top_text, bottom_text):
+    """밈 이미지 생성 함수"""
+    try:
+        if template_choice in MEME_TEMPLATES:
+            template_info = MEME_TEMPLATES[template_choice]
+            # 기본 템플릿 이미지 생성 (실제로는 미리 준비된 이미지 사용)
+            img = create_meme_image(template_info, top_text, bottom_text)
+            return img
+        else:
+            return "템플릿을 선택해주세요."
+    except Exception as e:
+        return f"밈 생성 중 오류가 발생했습니다: {str(e)}"
 
-# 1. pre-trained 모델 로드(사전학습된 Question Answering 파이프라인)
-qna = pipeline("question-answering")
+def chatbot_response(message, history):
+    """챗봇 응답 생성"""
+    try:
+        if not message.strip():
+            return "메시지를 입력해주세요."
+        
+        # 간단한 키워드 기반 응답
+        response = generate_response(message, qa_pipeline, text_generator)
+        return response
+        
+    except Exception as e:
+        return f"응답 생성 중 오류가 발생했습니다: {str(e)}"
 
-def answer_question(question, context):
-    result = qna({
-        "question": question,
-        "context": context
-    })
-    # 결과가 리스트면 첫 답변만, 아니면 answer값 반환
-    if isinstance(result, list):
-        return result[0]['answer']
-    return result['answer']
+def analyze_meme_trend(keyword):
+    """밈 트렌드 분석 (시뮬레이션)"""
+    trends = [
+        f"'{keyword}' 관련 밈이 최근 {random.randint(10, 50)}% 증가했습니다.",
+        f"'{keyword}' 밈의 인기도: {random.randint(70, 95)}/100",
+        f"유사한 키워드: {', '.join(random.sample(['재미있는', '웃긴', '인기', '바이럴', '트렌드'], 3))}"
+    ]
+    return "\n".join(trends)
 
-# 2. Gradio 인터페이스 정의 (입력창에 예시까지 기본값으로 넣기)
-demo = gr.Interface(
-    fn=answer_question,
-    inputs=[
-        gr.Textbox(
-            lines=2, 
-            label="질문 (Question)",
-            value="대한민국의 수도는 어디입니까?"  # ★ 질문 예시
-        ),
-        gr.Textbox(
-            lines=6, 
-            label="문맥 (Context)",
-            value="대한민국은 동아시아에 위치한 국가로, 수도는 서울이다. 서울은 정치, 경제, 문화의 중심지로 약 천만 명이 거주한다."  # ★ 문맥/지문 예시
-        )
-    ],
-    outputs=gr.Textbox(label="답변 (Answer)"),
-    title="질의응답(Q&A) 챗봇",
-    description="질문(Question)과 문맥(지문/Context)을 입력하면 AI가 빠르게 답변해줍니다."
-)
+# Gradio 인터페이스 구성
+with gr.Blocks(title=APP_CONFIG["title"], theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🎭 밈 생성 & AI 챗봇 서비스")
+    gr.Markdown("밈을 만들고, AI와 대화하며, 트렌드를 분석해보세요!")
+    
+    with gr.Tabs():
+        # 밈 생성 탭
+        with gr.TabItem("🖼️ 밈 생성기"):
+            with gr.Row():
+                with gr.Column():
+                    template_dropdown = gr.Dropdown(
+                        choices=list(MEME_TEMPLATES.keys()),
+                        label="밈 템플릿 선택",
+                        value=list(MEME_TEMPLATES.keys())[0]
+                    )
+                    top_text_input = gr.Textbox(
+                        label="상단 텍스트",
+                        placeholder="상단에 들어갈 텍스트를 입력하세요",
+                        value="When you see a bug"
+                    )
+                    bottom_text_input = gr.Textbox(
+                        label="하단 텍스트", 
+                        placeholder="하단에 들어갈 텍스트를 입력하세요",
+                        value="But it works in production"
+                    )
+                    create_btn = gr.Button("밈 생성하기", variant="primary")
+                
+                with gr.Column():
+                    meme_output = gr.Image(label="생성된 밈")
+            
+            create_btn.click(
+                fn=create_meme,
+                inputs=[template_dropdown, top_text_input, bottom_text_input],
+                outputs=meme_output
+            )
+        
+        # AI 챗봇 탭
+        with gr.TabItem("🤖 AI 챗봇"):
+            chatbot = gr.Chatbot(
+                label="AI 어시스턴트",
+                placeholder="안녕하세요! 무엇을 도와드릴까요?",
+                height=400
+            )
+            msg_input = gr.Textbox(
+                label="메시지 입력",
+                placeholder="궁금한 것을 물어보세요...",
+                lines=2
+            )
+            
+            with gr.Row():
+                send_btn = gr.Button("전송", variant="primary")
+                clear_btn = gr.Button("대화 초기화")
+            
+            def user_message(message, history):
+                return "", history + [[message, None]]
+            
+            def bot_message(history):
+                if history and history[-1][1] is None:
+                    user_msg = history[-1][0]
+                    bot_response = chatbot_response(user_msg, history)
+                    history[-1][1] = bot_response
+                return history
+            
+            msg_input.submit(user_message, [msg_input, chatbot], [msg_input, chatbot]).then(
+                bot_message, chatbot, chatbot
+            )
+            send_btn.click(user_message, [msg_input, chatbot], [msg_input, chatbot]).then(
+                bot_message, chatbot, chatbot
+            )
+            clear_btn.click(lambda: [], outputs=chatbot)
+        
+        # 트렌드 분석 탭
+        with gr.TabItem("📈 트렌드 분석"):
+            with gr.Row():
+                with gr.Column():
+                    keyword_input = gr.Textbox(
+                        label="키워드 입력",
+                        placeholder="분석할 키워드를 입력하세요",
+                        value="AI"
+                    )
+                    analyze_btn = gr.Button("트렌드 분석", variant="primary")
+                
+                with gr.Column():
+                    trend_output = gr.Textbox(
+                        label="분석 결과",
+                        lines=5,
+                        interactive=False
+                    )
+            
+            analyze_btn.click(
+                fn=analyze_meme_trend,
+                inputs=keyword_input,
+                outputs=trend_output
+            )
+    
+    gr.Markdown("---")
+    gr.Markdown("💡 **사용법**: 각 탭을 클릭하여 밈 생성, AI 채팅, 트렌드 분석 기능을 이용하세요!")
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False
+    )

config.py

@@ -0,0 +1,69 @@
+"""
+애플리케이션 설정 및 상수 정의
+"""
+
+# 앱 기본 설정
+APP_CONFIG = {
+    "title": "밈 생성 & AI 챗봇 서비스",
+    "description": "밈을 만들고 AI와 대화하는 올인원 서비스",
+    "version": "1.0.0"
+}
+
+# 밈 템플릿 정의
+MEME_TEMPLATES = {
+    "드레이크 밈": {
+        "description": "드레이크가 거부하고 승인하는 밈",
+        "image_path": "drake_template.jpg",
+        "text_positions": [(50, 100), (50, 300)]
+    },
+    "디스트랙티드 보이프렌드": {
+        "description": "다른 여자를 보는 남자친구 밈",
+        "image_path": "distracted_boyfriend.jpg", 
+        "text_positions": [(100, 50), (300, 50), (500, 50)]
+    },
+    "체인지 마이 마인드": {
+        "description": "내 마음을 바꿔봐 밈",
+        "image_path": "change_my_mind.jpg",
+        "text_positions": [(200, 150)]
+    },
+    "원 두스 낫 심플리": {
+        "description": "단순히 ~하지 않는다 밈",
+        "image_path": "one_does_not_simply.jpg",
+        "text_positions": [(150, 50), (150, 350)]
+    }
+}
+
+# 챗봇 기본 응답
+CHATBOT_RESPONSES = {
+    "greetings": [
+        "안녕하세요! 무엇을 도와드릴까요?",
+        "반갑습니다! 궁금한 것이 있으시면 언제든 물어보세요.",
+        "안녕하세요! 밈 생성이나 다른 도움이 필요하시면 말씀해주세요."
+    ],
+    "meme_help": [
+        "밈 생성기 탭에서 템플릿을 선택하고 텍스트를 입력하면 됩니다!",
+        "인기 있는 밈 템플릿들을 준비해두었어요. 창의적인 텍스트를 넣어보세요!",
+        "상단과 하단 텍스트를 입력하면 재미있는 밈이 만들어집니다."
+    ],
+    "default": [
+        "흥미로운 질문이네요! 더 자세히 설명해주시겠어요?",
+        "그것에 대해 더 알고 싶습니다. 구체적으로 어떤 부분이 궁금하신가요?",
+        "좋은 질문입니다! 어떤 도움이 필요하신지 말씀해주세요."
+    ]
+}
+
+# 색상 및 스타일 설정
+STYLE_CONFIG = {
+    "primary_color": "#FF6B6B",
+    "secondary_color": "#4ECDC4", 
+    "background_color": "#F7F9FC",
+    "text_color": "#2C3E50",
+    "font_family": "Arial, sans-serif"
+}
+
+# API 및 외부 서비스 설정
+API_CONFIG = {
+    "huggingface_api_url": "https://api-inference.huggingface.co/models/",
+    "timeout": 30,
+    "max_retries": 3
+}

utils.py

@@ -0,0 +1,180 @@
+"""
+유틸리티 함수들
+"""
+
+import random
+import requests
+from PIL import Image, ImageDraw, ImageFont
+import io
+from config import CHATBOT_RESPONSES, MEME_TEMPLATES
+
+def create_meme_image(template_info, top_text, bottom_text):
+    """밈 이미지 생성"""
+    try:
+        # 기본 캔버스 생성 (실제 환경에서는 템플릿 이미지 로드)
+        width, height = 500, 400
+        img = Image.new('RGB', (width, height), color='white')
+        draw = ImageDraw.Draw(img)
+        
+        # 폰트 설정 (기본 폰트 사용)
+        try:
+            font_large = ImageFont.truetype("arial.ttf", 32)
+            font_small = ImageFont.truetype("arial.ttf", 24)
+        except:
+            font_large = ImageFont.load_default()
+            font_small = ImageFont.load_default()
+        
+        # 배경 그라데이션 효과
+        for y in range(height):
+            color_value = int(255 * (y / height))
+            color = (color_value, color_value, 255)
+            draw.line([(0, y), (width, y)], fill=color)
+        
+        # 텍스트 추가
+        if top_text:
+            # 상단 텍스트
+            text_width = draw.textlength(top_text, font=font_large)
+            x = (width - text_width) // 2
+            y = 30
+            # 텍스트 외곽선
+            for dx in [-1, 0, 1]:
+                for dy in [-1, 0, 1]:
+                    if dx != 0 or dy != 0:
+                        draw.text((x+dx, y+dy), top_text, font=font_large, fill='black')
+            draw.text((x, y), top_text, font=font_large, fill='white')
+        
+        if bottom_text:
+            # 하단 텍스트
+            text_width = draw.textlength(bottom_text, font=font_large)
+            x = (width - text_width) // 2
+            y = height - 70
+            # 텍스트 외곽선
+            for dx in [-1, 0, 1]:
+                for dy in [-1, 0, 1]:
+                    if dx != 0 or dy != 0:
+                        draw.text((x+dx, y+dy), bottom_text, font=font_large, fill='black')
+            draw.text((x, y), bottom_text, font=font_large, fill='white')
+        
+        return img
+        
+    except Exception as e:
+        print(f"이미지 생성 오류: {e}")
+        return create_error_image(str(e))
+
+def create_error_image(error_msg):
+    """에러 이미지 생성"""
+    img = Image.new('RGB', (400, 300), color='red')
+    draw = ImageDraw.Draw(img)
+    try:
+        font = ImageFont.load_default()
+        draw.text((50, 150), f"오류: {error_msg}", font=font, fill='white')
+    except:
+        pass
+    return img
+
+def add_text_to_image(img, text, position, font_size=32, color='white'):
+    """이미지에 텍스트 추가"""
+    draw = ImageDraw.Draw(img)
+    try:
+        font = ImageFont.truetype("arial.ttf", font_size)
+    except:
+        font = ImageFont.load_default()
+    
+    x, y = position
+    # 외곽선 효과
+    for dx in [-1, 0, 1]:
+        for dy in [-1, 0, 1]:
+            if dx != 0 or dy != 0:
+                draw.text((x+dx, y+dy), text, font=font, fill='black')
+    draw.text((x, y), text, font=font, fill=color)
+    
+    return img
+
+def generate_response(message, qa_pipeline=None, text_generator=None):
+    """챗봇 응답 생성"""
+    message_lower = message.lower()
+    
+    # 인사말 처리
+    if any(greeting in message_lower for greeting in ['안녕', '하이', '헬로', '반가']):
+        return random.choice(CHATBOT_RESPONSES["greetings"])
+    
+    # 밈 관련 질문 처리
+    if any(word in message_lower for word in ['밈', 'meme', '이미지', '생성']):
+        return random.choice(CHATBOT_RESPONSES["meme_help"])
+    
+    # AI 모델을 사용한 응답 생성 시도
+    if qa_pipeline and len(message) > 10:
+        try:
+            # 간단한 컨텍스트로 QA 수행
+            context = "이것은 밈 생성과 AI 채팅 서비스입니다. 사용자들이 재미있는 밈을 만들고 AI와 대화할 수 있습니다."
+            result = qa_pipeline(question=message, context=context)
+            if result and 'answer' in result:
+                return f"🤖 {result['answer']}"
+        except Exception as e:
+            print(f"QA 파이프라인 오류: {e}")
+    
+    # 텍스트 생성 모델 사용 시도
+    if text_generator:
+        try:
+            result = text_generator(f"사용자: {message}\n봇:", max_length=50, num_return_sequences=1)
+            if result and len(result) > 0:
+                generated_text = result[0]['generated_text']
+                bot_response = generated_text.split('봇:')[-1].strip()
+                if bot_response and len(bot_response) > 5:
+                    return f"🤖 {bot_response}"
+        except Exception as e:
+            print(f"텍스트 생성 오류: {e}")
+    
+    # 기본 응답
+    return random.choice(CHATBOT_RESPONSES["default"])
+
+def validate_input(text, max_length=100):
+    """입력 텍스트 검증"""
+    if not text or not text.strip():
+        return False, "텍스트를 입력해주세요."
+    
+    if len(text) > max_length:
+        return False, f"텍스트가 너무 깁니다. {max_length}자 이하로 입력해주세요."
+    
+    # 특수문자 필터링 (선택적)
+    forbidden_chars = ['<', '>', '&', '"', "'"]
+    if any(char in text for char in forbidden_chars):
+        return False, "허용되지 않는 특수문자가 포함되어 있습니다."
+    
+    return True, "검증 통과"
+
+def get_random_meme_idea():
+    """랜덤 밈 아이디어 제공"""
+    ideas = [
+        ("When you finish your project", "But remember you didn't write tests"),
+        ("Me: I'll go to bed early tonight", "Also me at 3 AM:"),
+        ("CSS in theory", "CSS in practice"),
+        ("My code works", "I have no idea why"),
+        ("Expected behavior", "Actual behavior")
+    ]
+    return random.choice(ideas)
+
+def format_trend_data(keyword, data=None):
+    """트렌드 데이터 포맷팅"""
+    if not data:
+        # 시뮬레이션 데이터 생성
+        popularity = random.randint(60, 95)
+        growth = random.randint(-20, 50)
+        related_terms = random.sample([
+            '바이럴', '인기', '트렌드', '웃긴', '재미있는', 
+            '핫이슈', '화제', '밈', '패러디', '유머'
+        ], 3)
+        
+        result = f"""
+📊 '{keyword}' 트렌드 분석 결과
+
+🔥 인기도: {popularity}/100
+📈 성장률: {growth:+d}%
+🏷️ 관련 키워드: {', '.join(related_terms)}
+⏰ 분석 시간: 방금 전
+
+💡 추천: {'상승 트렌드입니다!' if growth > 0 else '관심을 끌 수 있는 키워드입니다.'}
+        """
+        return result.strip()
+    
+    return str(data)