Update app.py

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, jsonify, request
+import gradio as gr
 import threading
 import discord
 import logging
@@ -6,9 +6,6 @@ import os
 from huggingface_hub import InferenceClient
 import asyncio
 
-# 플라스크 애플리케이션 설정
-app = Flask(__name__)
-
 # 로깅 설정
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(name)s: %(message)s', handlers=[logging.StreamHandler()])
 
@@ -98,32 +95,39 @@ async def generate_response(user_input):
     conversation_history.append({"role": "assistant", "content": full_response_text})
     return full_response_text
 
-# 플라스크 라우트 설정
-@app.route('/')
-def index():
-    return "디스코드 봇이 실행 중입니다."
-
-@app.route('/status', methods=['GET'])
-def status():
-    return jsonify({"status": "봇이 실행 중입니다."})
-
-@app.route('/send-message', methods=['POST'])
-def send_message():
-    data = request.json
-    channel_id = data.get('channel_id')
-    message_content = data.get('message')
-    channel = discord_client.get_channel(channel_id)
+# Gradio 함수 정의
+def send_message_to_discord(channel_id, message_content):
+    channel = discord_client.get_channel(int(channel_id))
     if channel:
         asyncio.run_coroutine_threadsafe(channel.send(message_content), discord_client.loop)
-        return jsonify({"status": "메시지 전송 완료"}), 200
+        return "메시지 전송 완료"
     else:
-        return jsonify({"error": "유효하지 않은 채널 ID"}), 400
+        return "유효하지 않은 채널 ID"
+
+# Gradio 인터페이스 설정
+with gr.Blocks() as demo:
+    with gr.Row():
+        gr.Markdown("## 디스코드 봇 상태 및 메시지 전송")
+    with gr.Row():
+        status_button = gr.Button("상태 확인")
+        status_output = gr.Textbox(label="봇 상태", placeholder="상태 버튼을 클릭하세요.", interactive=False)
+    with gr.Row():
+        channel_id_input = gr.Textbox(label="채널 ID", placeholder="디스코드 채널 ID를 입력하세요.")
+        message_input = gr.Textbox(label="메시지 내용", placeholder="전송할 메시지 내용을 입력하세요.")
+        send_button = gr.Button("메시지 전송")
+        send_output = gr.Textbox(label="메시지 전송 결과", placeholder="메시지 전송 결과가 여기에 표시됩니다.", interactive=False)
+
+    def check_status():
+        return "봇이 실행 중입니다."
+
+    status_button.click(fn=check_status, outputs=status_output)
+    send_button.click(fn=send_message_to_discord, inputs=[channel_id_input, message_input], outputs=send_output)
 
+# 디스코드 봇을 별도의 스레드에서 실행
 def run_discord_bot():
     discord_client.run(os.getenv('DISCORD_TOKEN'))
 
-# 디스코드 봇을 별도의 스레드에서 실행
 if __name__ == "__main__":
     discord_client = MyClient(intents=intents)
     threading.Thread(target=run_discord_bot).start()  # 봇 실행
-    app.run(host='0.0.0.0', port=5000)  # 플라스크 서버 실행
+    demo.launch(server_name="0.0.0.0", server_port=5000)  # Gradio 서버 실행