Update app.py

app.py

@@ -2,25 +2,30 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-# 모델 및 토크나이저 로드
 MODEL_NAME = "lcw99/ko-dialoGPT-korean-chit-chat"
+
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
 
-def chatbot_response(input_text, history=[]):
-    """챗봇 응답 생성"""
-    input_ids = tokenizer.encode(input_text + tokenizer.eos_token, return_tensors="pt")
+def chat_with_ai(history, message):
+    input_text = message + tokenizer.eos_token
+    input_ids = tokenizer.encode(input_text, return_tensors="pt")
+
     response_ids = model.generate(input_ids, max_length=100, pad_token_id=tokenizer.eos_token_id)
     response_text = tokenizer.decode(response_ids[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
-    
-    # Gradio 최신 형식 적용
-    history.append({"role": "user", "content": input_text})
+
+    history.append({"role": "user", "content": message})
     history.append({"role": "assistant", "content": response_text})
-    return response_text, history
 
-# Gradio 인터페이스 설정
-demo = gr.ChatInterface(fn=chatbot_response, title="Korean Chatbot API", description="lcw99/ko-dialoGPT-korean-chit-chat 기반 챗봇")
+    return history, ""
+
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot(label="Ko-DialoGPT Chatbot", type="messages")
+    message = gr.Textbox(label="입력 메시지")
+    clear_btn = gr.Button("초기화")
 
-# ✅ API 활성화 설정
-demo.launch(server_name="0.0.0.0", server_port=7860, share=True)
+    message.submit(chat_with_ai, [chatbot, message], [chatbot, message])
+    clear_btn.click(lambda: [], [], chatbot)
 
+# ✅ 서버 포트 및 주소 추가
+demo.launch(server_name="0.0.0.0", server_port=7860)