update sth. at 2024-11-15 18:59:00

app.py

@@ -1,23 +1,19 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 from PIL import Image
-import googletrans  # 用于中文到英文的翻译
 
 # ===================== 核心逻辑模块 =====================
 
-# 初始化翻译客户端
-translator = googletrans.Translator()
-
 # 初始化模型客户端
 try:
     # 文本聊天模型
     client_text = InferenceClient("meta-llama/Llama-3.2-11B-Vision-Instruct")
     
-    # 图片生成模型
-    client_image = InferenceClient("black-forest-labs/FLUX.1-dev")
+    # 图片生成模型 1
+    client_image_1 = InferenceClient()
     
-    # 图像问答模型（暂时隐藏）
-    client_vqa = None
+    # 图片生成模型 2 (FLUX)
+    client_image_2 = InferenceClient("black-forest-labs/FLUX.1-dev")
 except Exception as e:
     print(f"Error initializing clients: {e}")
 
@@ -36,25 +32,19 @@ def chat_with_model(messages):
 # ---------- 图像生成模块 ----------
 def image_gen(prompt):
     """
-    调用图像生成模型生成图像，并翻译中文提示词为英文。
+    调用两个图像生成模型，生成两个图像。
     """
     try:
-        # 如果是中文，先翻译成英文
-        if any(ord(c) > 127 for c in prompt):  # 判断是否是中文字符
-            prompt = translator.translate(prompt, src='zh-cn', dest='en').text
-        
-        # 显示服务正在生成图像
-        status_message = "图像生成中，请稍候..."
-        
-        image = client_image.text_to_image(prompt)
+        # 使用服务一 (默认模型)
+        image_1 = client_image_1.text_to_image(prompt)
         
-        # 图像生成完成，更新状态
-        status_message = "图像生成完成！"
+        # 使用服务二 (FLUX 模型)
+        image_2 = client_image_2.text_to_image(prompt)
         
-        return image, prompt, status_message  # 返回图像和原始提示词及状态
+        return image_1, image_2  # 返回两个生成的图像
     except Exception as e:
         print(f"Image generation failed: {e}")
-        return None, "图像生成失败，请稍后再试。", "图像生成失败，请稍后再试。"  # 返回错误消息
+        return None, None  # 如果生成失败，返回两个空值
 
 # ===================== Gradio 界面构建 =====================
 
@@ -81,16 +71,16 @@ def build_interface():
         # 图像生成模块
         with gr.Tab("图像生成"):
             image_prompt = gr.Textbox(label="图像提示词", placeholder="描述你想生成的图像")
-            image_output = gr.Image(label="生成的图像")
-            image_description = gr.Textbox(label="提示词", placeholder="输入中文或英文提示词", interactive=False)
+            image_output_1 = gr.Image(label="服务一生成的图像")
+            image_output_2 = gr.Image(label="服务二生成的图像")
             image_button = gr.Button("生成图像")
 
             # 处理图像生成请求
             def image_handler(prompt):
-                img, translated_prompt, status = image_gen(prompt)
-                return img, translated_prompt, status
+                img_1, img_2 = image_gen(prompt)
+                return img_1, img_2
 
-            image_button.click(image_handler, inputs=image_prompt, outputs=[image_output, image_description, status_output])
+            image_button.click(image_handler, inputs=image_prompt, outputs=[image_output_1, image_output_2])
 
         gr.Markdown("### 使用说明")
         gr.Markdown("本助手支持文本聊天和图像生成功能，使用上方选项卡切换不同功能。")

requirements.txt

@@ -1,4 +1,3 @@
 gradio
 huggingface_hub
-googletrans
 Pillow