111

scripts/app.py

@@ -0,0 +1,67 @@
+import logging
+import threading
+from threading import Lock
+from fastapi import FastAPI
+from backend_lsnet.inference import *
+from backend_lsnet.ui import *
+from backend_lsnet.api import Api
+import uvicorn
+import gradio as gr
+import os
+import argparse
+
+logging.basicConfig(level=logging.INFO)
+
+def parse_args():
+    parser = argparse.ArgumentParser(description="LSNet Artist Inference WebUI")
+    parser.add_argument("--host", type=str, default="127.0.0.1", help="Server host")
+    parser.add_argument("--port", type=int, default=7860, help="Server port")
+    return parser.parse_args()
+
+try:
+    from modules import script_callbacks, shared
+    IN_WEBUI = True
+except ImportError:
+    IN_WEBUI = False
+    shared = type('Shared', (), {'opts': type('Opts', (), {
+        'outdir_samples': '',
+        'outdir_txt2img_samples': '',
+        'outdir_img2img_samples': ''
+    })})()
+
+if IN_WEBUI:
+    from backend_lsnet.api import on_app_started
+    def on_ui_tabs():
+        block = create_ui()
+        return [(block, "LSNet Artist", "lsnet_tab")]
+    script_callbacks.on_ui_tabs(on_ui_tabs)
+    script_callbacks.on_app_started(on_app_started)
+else:
+    if __name__ == "__main__":
+        args = parse_args()
+        # Create models directory
+        os.makedirs("models/lsnet", exist_ok=True)
+        
+        app = FastAPI(docs_url="/docs", openapi_url="/openapi.json")
+        queue_lock = Lock()
+        api = Api(app, queue_lock, prefix="/lsnet/v1")
+        logging.info("API 路由已挂载到 FastAPI 实例")
+
+        block = create_ui()
+        logging.info("Gradio 界面已创建")
+
+        # Mount Gradio app to FastAPI
+        app = gr.mount_gradio_app(app, block, path="")
+
+        print(f"应用启动在：http://{args.host}:{args.port}")
+        print(f"API 文档：http://{args.host}:{args.port}/docs")
+
+        try:
+            uvicorn.run(
+                app,
+                host=args.host,
+                port=args.port,
+                log_level="info"
+            )
+        except Exception as e:
+            logging.error(f"启动失败: {str(e)}")