Update app.py

app.py

@@ -12,14 +12,22 @@ CAMERA_ANGLES = ["low-angle shot", "close-up shot", "bird's-eye view", "wide-ang
 QUALITY_PROMPTS = ["8k", "ultra-realistic", "high detail", "cinematic lighting", "award-winning"]
 
 # ========== 工具函数 ==========
-def load_candidates_from_file(file):
+def load_candidates_from_files(files):
     """
-    从上传的文件中读取候选项。
+    从多个文件中加载候选项。
     """
-    if file is None:
-        return []
-    lines = file.decode("utf-8").split("\n")
-    return [line.strip() for line in lines if line.strip()]
+    all_lines = []
+    if files:
+        for file in files:
+            # 如果 `file` 是路径
+            if isinstance(file, str):
+                with open(file, "r", encoding="utf-8") as f:
+                    all_lines.extend([line.strip() for line in f if line.strip()])
+            # 如果 `file` 是 Gradio 的文件对象
+            elif hasattr(file, "name"):
+                with open(file.name, "r", encoding="utf-8") as f:
+                    all_lines.extend([line.strip() for line in f if line.strip()])
+    return all_lines
 
 def get_random_item(candidates):
     """
@@ -44,7 +52,7 @@ def generate_natural_language_description(tags, api_key=None):
                     {"role": "user", "content": f"Here are the tags: {tags}\nPlease generate a vivid, imaginative scene description."}
                 ],
                 max_tokens=300,
-                temperature=0.8,
+                temperature=0.7,
             )
             return response["choices"][0]["message"]["content"].strip()
         except Exception as e:
@@ -59,15 +67,15 @@ def generate_natural_language_description(tags, api_key=None):
             f"with a {tags.get('expression', 'neutral expression')}, holding {tags.get('items', 'something')} among {tags.get('other_details', 'subtle details')}."
         )
 
-def generate_prompt(action_file, style_file, artist_file, character_file, api_key, selected_categories):
+def generate_prompt(action_file, style_file, artist_files, character_files, api_key, selected_categories):
     """
     生成随机提示词和描述。
     """
     # 从文件加载候选项
-    actions = load_candidates_from_file(action_file) if action_file else []
-    styles = load_candidates_from_file(style_file) if style_file else []
-    artists = load_candidates_from_file(artist_file) if artist_file else []
-    characters = load_candidates_from_file(character_file) if character_file else []
+    actions = load_candidates_from_files([action_file]) if action_file else []
+    styles = load_candidates_from_files([style_file]) if style_file else []
+    artists = load_candidates_from_files(artist_files) if artist_files else []
+    characters = load_candidates_from_files(character_files) if character_files else []
 
     # 确定角色类型
     number_of_characters = ", ".join(selected_categories) if selected_categories else random.choice(["1girl", "1boy"])
@@ -114,8 +122,10 @@ def gradio_interface():
         with gr.Row():
             action_file = gr.File(label="Upload Action File (Optional)", file_types=[".txt"])
             style_file = gr.File(label="Upload Style File (Optional)", file_types=[".txt"])
-            artist_file = gr.File(label="Upload Artist File (Optional)", file_types=[".txt"])
-            character_file = gr.File(label="Upload Character File (Optional)", file_types=[".txt"])
+        
+        with gr.Row():
+            artist_files = gr.File(label="Upload Artist Files (Multiple Allowed)", file_types=[".txt"], file_types_description="Text Files", multiple=True)
+            character_files = gr.File(label="Upload Character Files (Multiple Allowed)", file_types=[".txt"], file_types_description="Text Files", multiple=True)
 
         # 角色类型选择
         selected_categories = gr.CheckboxGroup(
@@ -134,7 +144,7 @@ def gradio_interface():
         # 按钮动作
         generate_button.click(
             generate_prompt,
-            inputs=[action_file, style_file, artist_file, character_file, api_key_input, selected_categories],
+            inputs=[action_file, style_file, artist_files, character_files, api_key_input, selected_categories],
             outputs=[tags_output, description_output],
         )
 
@@ -142,4 +152,5 @@ def gradio_interface():
 
 # 启动 Gradio 应用
 if __name__ == "__main__":
-    gradio_interface().launch()
+    gradio_interface().launch(share=True)
+