Update app.py

app.py

@@ -56,7 +56,7 @@ def plot_feature_differences(latent_diff):
 
     plt.figure(figsize=(8, 4))
     plt.bar(indices, diff_magnitude, alpha=0.7)
-    plt.xlabel("Feature Index")
+    plt.xlabel("Feature Index (Latent Dimension)")
     plt.ylabel("Magnitude of Difference")
     plt.title("Feature Differences (Bar Chart)")
     bar_chart_path = "bar_chart.png"
@@ -64,7 +64,7 @@ def plot_feature_differences(latent_diff):
     plt.close()
 
     plt.figure(figsize=(6, 6))
-    plt.pie(diff_magnitude[:10], labels=range(10), autopct="%1.1f%%", startangle=140)
+    plt.pie(diff_magnitude[:10], labels=[f"Feature {i}" for i in range(10)], autopct="%1.1f%%", startangle=140)
     plt.title("Top 10 Feature Differences (Pie Chart)")
     pie_chart_path = "pie_chart.png"
     plt.savefig(pie_chart_path)
@@ -120,13 +120,7 @@ def analyze_images(img_a, img_b, api_key, api_type):
     }
 
 # 批量分析
-def batch_analyze(folder_a, folder_b, api_key, api_type):
-    def load_images(folder_path):
-        files = sorted([os.path.join(folder_path, f) for f in os.listdir(folder_path) if f.lower().endswith(('.png', '.jpg', '.jpeg'))])
-        return [Image.open(f).convert("RGB") for f in files]
-
-    images_a = load_images(folder_a)
-    images_b = load_images(folder_b)
+def batch_analyze(images_a, images_b, api_key, api_type):
     num_pairs = min(len(images_a), len(images_b))
 
     results = []
@@ -143,17 +137,20 @@ with gr.Blocks() as demo:
     gr.Markdown("# 批量图像对比分析工具")
 
     api_key_input = gr.Textbox(label="API Key", placeholder="输入您的 API Key", type="password")
-    api_type_input = gr.Dropdown(label="API 类型", choices=["GPT", "DeepSeek"], value="GPT")
-    folder_a_input = gr.Textbox(label="文件夹A路径", placeholder="输入包含图片A的文件夹路径")
-    folder_b_input = gr.Textbox(label="文件夹B路径", placeholder="输入包含图片B的文件夹路径")
+    api_type_input = gr.Radio(label="API 类型", choices=["GPT", "DeepSeek"], value="GPT")
+    images_a_input = gr.File(label="上传文件夹A图片", file_types=[".png", ".jpg", ".jpeg"], file_count="multiple")
+    images_b_input = gr.File(label="上传文件夹B图片", file_types=[".png", ".jpg", ".jpeg"], file_count="multiple")
     analyze_button = gr.Button("开始批量分析")
 
     with gr.Row():
         result_gallery = gr.Gallery(label="差异图像")
     result_text_analysis = gr.Textbox(label="详细分析", interactive=False, lines=5)
 
-    def process_batch_analysis(folder_a, folder_b, api_key, api_type):
-        results = batch_analyze(folder_a, folder_b, api_key, api_type)
+    def process_batch_analysis(images_a, images_b, api_key, api_type):
+        images_a = [Image.open(img).convert("RGB") for img in images_a]
+        images_b = [Image.open(img).convert("RGB") for img in images_b]
+        results = batch_analyze(images_a, images_b, api_key, api_type)
+
         all_images = []
         all_texts = []
 
@@ -167,7 +164,7 @@ with gr.Blocks() as demo:
 
     analyze_button.click(
         fn=process_batch_analysis,
-        inputs=[folder_a_input, folder_b_input, api_key_input, api_type_input],
+        inputs=[images_a_input, images_b_input, api_key_input, api_type_input],
         outputs=[result_gallery, result_text_analysis]
     )