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 서버 실행