Upload 4 files

app.py

@@ -1,56 +1,82 @@
 import gradio as gr
 from transformers import pipeline
+import random
+from PIL import Image, ImageDraw, ImageFont
+import io
 
-# 모델 로드
-qna = pipeline("question-answering")
+# 모델 초기화
+try:
+    qa_pipeline = pipeline("question-answering")
+except:
+    qa_pipeline = None
 
-def answer_question(question, context):
+def create_meme(template_choice, top_text, bottom_text):
+    """밈 이미지 생성"""
     try:
-        result = qna({
-            "question": question,
-            "context": context
-        })
-        if isinstance(result, list):
-            return result[0]['answer']
-        return result['answer']
+        # 간단한 밈 이미지 생성
+        width, height = 500, 400
+        img = Image.new('RGB', (width, height), color='lightblue')
+        draw = ImageDraw.Draw(img)
+        
+        # 텍스트 추가
+        if top_text:
+            draw.text((50, 50), top_text, fill='black')
+        if bottom_text:
+            draw.text((50, 350), bottom_text, fill='black')
+            
+        return img
     except Exception as e:
-        return f"오류가 발생했습니다: {str(e)}"
+        return f"오류: {str(e)}"
+
+def chatbot_response(message, history):
+    """챗봇 응답"""
+    responses = [
+        "흥미로운 질문이네요!",
+        "더 자세히 말씀해주세요.",
+        "그것에 대해 생각해보겠습니다.",
+        "좋은 아이디어입니다!"
+    ]
+    return random.choice(responses)
 
-# Gradio 인터페이스 생성
-with gr.Blocks() as demo:
-    gr.Markdown("# 질의응답 챗봇")
+# Gradio 인터페이스
+with gr.Blocks(title="밈 생성 & 챗봇") as demo:
+    gr.Markdown("# 🎭 밈 생성 & AI 챗봇 서비스")
     
-    with gr.Row():
-        with gr.Column():
-            question_input = gr.Textbox(
-                label="질문",
-                placeholder="질문을 입력하세요",
-                lines=2
-            )
-            context_input = gr.Textbox(
-                label="문맥",
-                placeholder="관련 문맥을 입력하세요",
-                lines=5
-            )
-            submit_btn = gr.Button("답변 생성", variant="primary")
+    with gr.Tabs():
+        # 밈 생성 탭
+        with gr.TabItem("🖼️ 밈 생성기"):
+            with gr.Row():
+                with gr.Column():
+                    template_dropdown = gr.Dropdown(
+                        choices=["드레이크 밈", "디스트랙티드 보이프렌드", "원 두스 낫 심플리"],
+                        label="밈 템플릿 선택",
+                        value="드레이크 밈"
+                    )
+                    top_text = gr.Textbox(label="상단 텍스트", value="When you see a bug")
+                    bottom_text = gr.Textbox(label="하단 텍스트", value="But it works in production")
+                    create_btn = gr.Button("밈 생성하기", variant="primary")
+                
+                with gr.Column():
+                    meme_output = gr.Image(label="생성된 밈")
+            
+            create_btn.click(create_meme, [template_dropdown, top_text, bottom_text], meme_output)
         
-        with gr.Column():
-            answer_output = gr.Textbox(
-                label="답변",
-                lines=3,
-                interactive=False
-            )
-    
-    submit_btn.click(
-        fn=answer_question,
-        inputs=[question_input, context_input],
-        outputs=answer_output
-    )
+        # 챗봇 탭
+        with gr.TabItem("🤖 AI 챗봇"):
+            chatbot = gr.Chatbot(label="AI 어시스턴트", height=400)
+            msg = gr.Textbox(label="메시지 입력", placeholder="궁금한 것을 물어보세요...")
+            
+            def user_msg(message, history):
+                return "", history + [[message, None]]
+            
+            def bot_msg(history):
+                if history and history[-1][1] is None:
+                    user_message = history[-1][0]
+                    bot_response = chatbot_response(user_message, history)
+                    history[-1][1] = bot_response
+                return history
+            
+            msg.submit(user_msg, [msg, chatbot], [msg, chatbot]).then(bot_msg, chatbot, chatbot)
 
-# 중요: Spaces 배포시 이 부분이 핵심
 if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",  # 모든 IP에서 접근 가능
-        server_port=7860,       # 기본 포트
-        share=False            # Spaces에서는 False 사용
-    )
+    demo.launch()