Update app.py

app.py

@@ -11,8 +11,10 @@ from pydantic import BaseModel
 from sentence_transformers import SentenceTransformer
 from typing import Optional, List,Dict
 from transformers import BertTokenizer, BertForSequenceClassification
-
-
+import uvicorn
+import gradio as gr
+from threading import Thread
+from fastapi.middleware.cors import CORSMiddleware
 
 #####################################
 # 1) 앱 및 모델 초기화
@@ -20,6 +22,13 @@ from transformers import BertTokenizer, BertForSequenceClassification
 app = FastAPI()
 import os
 
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # 모든 도메인 허용 (보안상 필요에 따라 제한)
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 os.environ["HF_HOME"] = "/workspace/huggingface_cache"
 os.environ["HF_HUB_CACHE"] = "/workspace/huggingface_cache"
 os.environ["TRANSFORMERS_CACHE"] = "/workspace/huggingface_cache"
@@ -1300,3 +1309,20 @@ def chat_or_recommend(req: ChatOrRecommendRequest):
 
     return response_dict
 
+def run_fastapi():
+    uvicorn.run(app, host="0.0.0.0", port=7860)
+
+def chatbot_interface(user_input, mode):
+    return chat_response(user_input, mode)
+
+iface = gr.Interface(
+    fn=chatbot_interface,
+    inputs=["text", gr.Radio(["emotion", "rational"], label="Mode")],
+    outputs="text",
+    title="Chatbot Interface",
+    description="FastAPI 기반 챗봇 + 추천 시스템"
+)
+
+if __name__ == "__main__":
+    Thread(target=run_fastapi).start()
+    iface.launch(server_name="0.0.0.0", server_port=7861)