Update app.py

app.py

@@ -1,14 +1,14 @@
 import discord
 import logging
 import os
-import subprocess
 from transformers import PaliGemmaForConditionalGeneration, PaliGemmaProcessor
 import torch
 import re
 import requests
 from PIL import Image
 import io
-import asyncio
+import gradio as gr
+import threading
 
 # 로깅 설정
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(name)s: %(message)s', handlers=[logging.StreamHandler()])
@@ -55,6 +55,45 @@ async def create_captions_rich(image: Image.Image) -> str:
     modified_caption = modify_caption(decoded)
     return modified_caption
 
+# Gradio 인터페이스 설정
+def create_captions_rich_sync(image):
+    return asyncio.run(create_captions_rich(image))
+
+css = """
+  #mkd {
+    height: 500px; 
+    overflow: auto; 
+    border: 1px solid #ccc; 
+  }
+"""
+
+with gr.Blocks(css=css) as demo:
+    gr.HTML("<h1><center>PaliGemma Fine-tuned for Long Captioning<center><h1>")
+    with gr.Tab(label="PaliGemma Long Captioner"):
+        with gr.Row():
+            with gr.Column():
+                input_img = gr.Image(label="Input Picture")
+                submit_btn = gr.Button(value="Submit")
+            output = gr.Text(label="Caption")
+
+        gr.Examples(
+            [["image1.jpg"], ["image2.jpg"], ["image3.png"], ["image4.jpg"], ["image5.jpg"], ["image6.PNG"]],
+            inputs=[input_img],
+            outputs=[output],
+            fn=create_captions_rich_sync,
+            label='Try captioning on examples'
+        )
+
+        submit_btn.click(create_captions_rich_sync, [input_img], [output])
+
+# Gradio 서버를 비동기적으로 실행
+def run_gradio():
+    demo.launch(
+        server_name="0.0.0.0", 
+        server_port=int(os.getenv("GRADIO_SERVER_PORT", 7861)), 
+        inbrowser=True
+    )
+
 # 특정 채널 ID 설정
 SPECIFIC_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID", "123456789012345678"))  # 환경 변수 또는 직접 설정
 
@@ -66,11 +105,8 @@ class MyClient(discord.Client):
 
     async def on_ready(self):
         logging.info(f'{self.user}로 로그인되었습니다!')
-        asyncio.create_task(self.start_gradio_server())
-        logging.info("Web.py 서버가 시작되었습니다.")
-
-    async def start_gradio_server(self):
-        subprocess.run(["python", "web.py"], check=True)
+        threading.Thread(target=run_gradio, daemon=True).start()
+        logging.info("Gradio 서버가 시작되었습니다.")
 
     async def on_message(self, message):
         if message.author == self.user:
@@ -106,3 +142,6 @@ async def download_image(url):
 if __name__ == "__main__":
     discord_client = MyClient(intents=intents)
     discord_client.run(os.getenv('DISCORD_TOKEN'))
+
+    discord_client = MyClient(intents=intents)
+    discord_client.run(os.getenv('DISCORD_TOKEN'))