Spaces:

Manggomee
/

test

Sleeping

test / app.py

Upload 4 files

460508b verified over 1 year ago

3.12 kB

import gradio as gr
from transformers import pipeline
import random
from PIL import Image, ImageDraw, ImageFont
import io

# 모델 초기화
try:
    qa_pipeline = pipeline("question-answering")
except:
    qa_pipeline = None

def create_meme(template_choice, top_text, bottom_text):
    """밈 이미지 생성"""
    try:
        # 간단한 밈 이미지 생성
        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)}"

def chatbot_response(message, history):
    """챗봇 응답"""
    responses = [
        "흥미로운 질문이네요!",
        "더 자세히 말씀해주세요.",
        "그것에 대해 생각해보겠습니다.",
        "좋은 아이디어입니다!"
    ]
    return random.choice(responses)

# Gradio 인터페이스
with gr.Blocks(title="밈 생성 & 챗봇") as demo:
    gr.Markdown("# 🎭 밈 생성 & AI 챗봇 서비스")
    
    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.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)

if __name__ == "__main__":
    demo.launch()

	import gradio as gr
	from transformers import pipeline
	import random
	from PIL import Image, ImageDraw, ImageFont
	import io

	# 모델 초기화
	try:
	qa_pipeline = pipeline("question-answering")
	except:
	qa_pipeline = None

	def create_meme(template_choice, top_text, bottom_text):
	"""밈 이미지 생성"""
	try:
	# 간단한 밈 이미지 생성
	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)}"

	def chatbot_response(message, history):
	"""챗봇 응답"""
	responses = [
	"흥미로운 질문이네요!",
	"더 자세히 말씀해주세요.",
	"그것에 대해 생각해보겠습니다.",
	"좋은 아이디어입니다!"
	]
	return random.choice(responses)

	# Gradio 인터페이스
	with gr.Blocks(title="밈 생성 & 챗봇") as demo:
	gr.Markdown("# 🎭 밈 생성 & AI 챗봇 서비스")

	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.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)

	if __name__ == "__main__":
	demo.launch()