Update app.py

app.py

@@ -1,7 +1,11 @@
 import gradio as gr
 import random
-import openai
 import os
+from openai import OpenAI
+from dotenv import load_dotenv
+
+# 加载环境变量
+load_dotenv()
 
 # ========== 默认选项和数据 ==========
 EXPRESSIONS = ["smiling", "determined", "surprised", "serene"]
@@ -35,31 +39,36 @@ def generate_natural_language_description(tags, api_key=None):
     """
     使用 OpenAI GPT 生成自然语言描述（适配 GPT-4o）。
     """
-    if api_key:
-        try:
-            openai.api_key = api_key.strip()
-            response = openai.ChatCompletion.create(
-                model="gpt-4",
-                messages=[
-                    {"role": "system", "content": (
+    # 如果用户未输入 API Key，则尝试从环境变量获取
+    if not api_key:
+        api_key = os.getenv("OPENAI_API_KEY")
+    if not api_key:
+        return "Error: No API Key provided and none found in environment variables."
+
+    try:
+        # 初始化 OpenAI 客户端
+        client = OpenAI(api_key=api_key)
+
+        # 调用 chat.completions.create
+        response = client.chat.completions.create(
+            messages=[
+                {
+                    "role": "system",
+                    "content": (
                         "You are a helpful assistant that generates creative painting/prompt descriptions. "
                         "Write at least three sentences in English, separated by periods."
-                    )},
-                    {"role": "user", "content": f"Here are the tags: {tags}\nPlease generate a vivid, imaginative scene description."}
-                ]
-            )
-            return response["choices"][0]["message"]["content"].strip()
-        except Exception as e:
-            return f"GPT generation failed. Error: {e}"
-    else:
-        # 本地逻辑生成
-        return (
-            f"In this scene, {tags.get('character_name', 'the character')} appears in a {tags.get('scene', 'mysterious place')}, "
-            f"captured from a {tags.get('camera_angle', 'unique angle')} perspective. "
-            f"The overall style combines {tags.get('artist_prompt', 'an unknown artist')} and {tags.get('style', 'a distinctive aesthetic')}, "
-            f"bringing a captivating atmosphere to the artwork. They are {tags.get('action', 'doing something')} "
-            f"with a {tags.get('expression', 'neutral expression')}, holding {tags.get('items', 'something')} among {tags.get('other_details', 'subtle details')}."
+                    ),
+                },
+                {
+                    "role": "user",
+                    "content": f"Here are the tags: {tags}\nPlease generate a vivid, imaginative scene description.",
+                },
+            ],
+            model="gpt-4o",
         )
+        return response["choices"][0]["message"]["content"].strip()
+    except Exception as e:
+        return f"GPT generation failed. Error: {e}"
 
 def generate_prompt(action_file, style_file, artist_files, character_files, api_key, selected_categories):
     """