Update app.py

app.py

@@ -5,6 +5,7 @@ from transformers import CLIPProcessor, CLIPModel, BlipProcessor, BlipForConditi
 import torch
 import matplotlib.pyplot as plt
 import numpy as np
+from openai import OpenAI
 
 # 初始化模型
 clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
@@ -75,13 +76,10 @@ def plot_feature_differences(latent_diff):
 
 # 生成详细分析
 def generate_text_analysis(api_key, api_type, caption_a, caption_b):
-    import openai
-
     if api_type == "DeepSeek":
-        from openai import OpenAI
         client = OpenAI(api_key=api_key, base_url="https://api.deepseek.com")
     else:
-        client = openai
+        client = OpenAI(api_key=api_key)
 
     response = client.chat.completions.create(
         model="gpt-4" if api_type == "GPT" else "deepseek-chat",
@@ -90,7 +88,8 @@ def generate_text_analysis(api_key, api_type, caption_a, caption_b):
             {"role": "user", "content": f"图片A的描述为：{caption_a}。图片B的描述为：{caption_b}。\n请对两张图片的内容和潜在特征区别进行详细分析，并输出一个简洁但富有条理的总结。"}
         ]
     )
-    return response['choices'][0]['message']['content'].strip()
+    # 修复: 正确访问返回值
+    return response.choices[0].message.content.strip()
 
 # 分析函数
 def analyze_images(img_a, img_b, api_key, api_type):