Update app.py

app.py

@@ -7,6 +7,7 @@ import modelscope_studio.components.pro as pro
 from openai import OpenAI
 from config import API_KEY, MODEL, SYSTEM_PROMPT, ENDPOINT, EXAMPLES, DEFAULT_LOCALE, DEFAULT_THEME
 from huggingface_hub import InferenceClient
+import os
 
 react_imports = {
     "lucide-react": "https://esm.sh/lucide-react@0.525.0",
@@ -68,8 +69,14 @@ class GradioEvents:
         messages.append({'role': 'user', 'content': input_value})
 
         try:
+            # Get API key from environment variable if not already set
+            api_key = API_KEY or os.getenv("HF_TOKEN") or os.getenv("OPENAI_API_KEY")
+            
+            if not api_key:
+                raise ValueError("No API key found. Please set HF_TOKEN or OPENAI_API_KEY environment variable.")
+            
             # Initialize HuggingFace InferenceClient
-            hf_client = InferenceClient(model=MODEL, token=API_KEY)
+            hf_client = InferenceClient(model=MODEL, token=api_key)
             
             # Try using HuggingFace InferenceClient first
             formatted_messages = [{"role": msg["role"], "content": msg["content"]} for msg in messages]
@@ -124,8 +131,14 @@ export default Demo
         except Exception as e:
             # Fallback to OpenAI client if HuggingFace client fails
             try:
+                # Get API key from environment variable if not already set
+                api_key = API_KEY or os.getenv("HF_TOKEN") or os.getenv("OPENAI_API_KEY")
+                
+                if not api_key:
+                    raise ValueError("No API key found. Please set HF_TOKEN or OPENAI_API_KEY environment variable.")
+                
                 # Initialize OpenAI client
-                openai_client = OpenAI(api_key=API_KEY, base_url=ENDPOINT + MODEL)
+                openai_client = OpenAI(api_key=api_key, base_url=ENDPOINT + MODEL)
                 
                 generator = openai_client.chat.completions.create(model=MODEL,
                                                                messages=messages,
@@ -446,7 +459,7 @@ with gr.Blocks(css=css) as demo:
                         antd.Tour.Step(
                             title="Step 3",
                             description="Wait for the result.",
-                            get_target=
+                        get_target=
                             "() => document.querySelector('#output-container')"
                         )
                         antd.Tour.Step(