update sth. at 2024-11-15 19:02:05

app.py

@@ -14,8 +14,12 @@ try:
     
     # 图片生成模型 2 (FLUX)
     client_image_2 = InferenceClient("black-forest-labs/FLUX.1-dev")
+    
+    # 更新状态为服务已启动
+    service_status = "服务已启动，您可以开始使用！"
 except Exception as e:
     print(f"Error initializing clients: {e}")
+    service_status = "服务初始化失败，请稍后再试。"
 
 # ---------- 文本聊天模块 ----------
 def chat_with_model(messages):
@@ -54,7 +58,7 @@ def build_interface():
     """
     with gr.Blocks() as demo:
         # 服务状态显示区域
-        status_output = gr.Textbox(label="服务状态", value="服务初始化中...", interactive=False)
+        status_output = gr.Textbox(label="服务状态", value=service_status, interactive=False)
         
         # 文本聊天模块
         with gr.Tab("文本聊天"):
@@ -71,8 +75,8 @@ def build_interface():
         # 图像生成模块
         with gr.Tab("图像生成"):
             image_prompt = gr.Textbox(label="图像提示词", placeholder="描述你想生成的图像")
-            image_output_1 = gr.Image(label="服务一生成的图像")
-            image_output_2 = gr.Image(label="服务二生成的图像")
+            image_output_1 = gr.Image(label="服务一生成的图像", elem_id="image_1", interactive=True).style(width=320, height=320)
+            image_output_2 = gr.Image(label="服务二生成的图像", elem_id="image_2", interactive=True).style(width=320, height=320)
             image_button = gr.Button("生成图像")
 
             # 处理图像生成请求