Upload 4 files

app.py

@@ -1,158 +1,56 @@
 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
+# 모델 로드
+qna = pipeline("question-answering")
 
-def create_meme(template_choice, top_text, bottom_text):
-    """밈 이미지 생성 함수"""
+def answer_question(question, context):
     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 "템플릿을 선택해주세요."
+        result = qna({
+            "question": question,
+            "context": context
+        })
+        if isinstance(result, list):
+            return result[0]['answer']
+        return result['answer']
     except Exception as e:
-        return f"밈 생성 중 오류가 발생했습니다: {str(e)}"
+        return f"오류가 발생했습니다: {str(e)}"
 
-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 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)
-
-# Gradio 인터페이스 구성
-with gr.Blocks(title=APP_CONFIG["title"], theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🎭 밈 생성 & AI 챗봇 서비스")
-    gr.Markdown("밈을 만들고, AI와 대화하며, 트렌드를 분석해보세요!")
+# Gradio 인터페이스 생성
+with gr.Blocks() as demo:
+    gr.Markdown("# 질의응답 챗봇")
     
-    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="궁금한 것을 물어보세요...",
+    with gr.Row():
+        with gr.Column():
+            question_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
+            context_input = gr.Textbox(
+                label="문맥",
+                placeholder="관련 문맥을 입력하세요",
+                lines=5
             )
-            send_btn.click(user_message, [msg_input, chatbot], [msg_input, chatbot]).then(
-                bot_message, chatbot, chatbot
-            )
-            clear_btn.click(lambda: [], outputs=chatbot)
+            submit_btn = gr.Button("답변 생성", variant="primary")
         
-        # 트렌드 분석 탭
-        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
+        with gr.Column():
+            answer_output = gr.Textbox(
+                label="답변",
+                lines=3,
+                interactive=False
             )
     
-    gr.Markdown("---")
-    gr.Markdown("💡 **사용법**: 각 탭을 클릭하여 밈 생성, AI 채팅, 트렌드 분석 기능을 이용하세요!")
+    submit_btn.click(
+        fn=answer_question,
+        inputs=[question_input, context_input],
+        outputs=answer_output
+    )
 
+# 중요: Spaces 배포시 이 부분이 핵심
 if __name__ == "__main__":
     demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=False
+        server_name="0.0.0.0",  # 모든 IP에서 접근 가능
+        server_port=7860,       # 기본 포트
+        share=False            # Spaces에서는 False 사용
     )

requirements.txt

@@ -1,3 +1,6 @@
 gradio==5.35.0
-transformers
-torch
+transformers==4.35.0
+torch==2.1.0
+Pillow==10.0.0
+requests==2.31.0
+numpy==1.24.3