Update app.py

app.py

@@ -7,6 +7,26 @@ from common.log import logger
 from plugins import *
 import signal
 import sys
+import gradio as gr
+from io import BytesIO
+from PIL import Image
+from concurrent.futures import ThreadPoolExecutor
+thread_pool = ThreadPoolExecutor(max_workers=8)
+
+def getImage(bytes):
+    bytes_stream = BytesIO(bytes)
+    image = Image.open(bytes_stream)
+    return image
+
+def getLoginUrl():
+    # load config
+    config.load_config()
+    # create channel
+    bot = channel_factory.create_channel("wx")
+    thread_pool.submit(bot.startup)
+    while (True):
+            if bot.getQrCode():
+                return getImage(bot.getQrCode())
 
 def sigterm_handler_wrap(_signo):
     old_handler = signal.getsignal(_signo)
@@ -42,4 +62,20 @@ def run():
         logger.exception(e)
 
 if __name__ == '__main__':
-    run()
+    #run()
+    try:
+        
+        with gr.Blocks() as demo:
+            with gr.Row():
+                with gr.Column():
+                    btn = gr.Button(value="生成二维码")
+                with gr.Column():
+                    outputs=[gr.Pil()]
+                btn.click(getLoginUrl, outputs=outputs)
+
+        demo.launch()
+
+
+    except Exception as e:
+        logger.error("App startup failed!")
+        logger.exception(e)