Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import openai
-from openai import OpenAI
-# 确保你的 OpenAI API 密钥
+from openai import OpenAI, OpenAIError
+
 
 
 def respond(message, history, system_message, max_tokens, temperature, top_p):
@@ -18,20 +18,24 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
       api_key="sk-Eqk19o6qqgk3i09tUkriFcESxqsQNsvdF18AaHaoxeHIv0oc",  
       base_url="https://api.moonshot.cn/v1", 
 ) 
-    response = client.ChatCompletion.create(
-        model="moonshot-v1-128k",
-        messages=messages,
-        max_tokens=max_tokens,
-        temperature=temperature,
-        top_p=top_p,
-        stream=True
-    )
-
-    final_response = ""
-    for message in response['choices'][0]['message']:
-        final_response += message['content']
-
-    return final_response
+    try:
+        response = client.ChatCompletion.create(
+            model="moonshot-v1-128k",
+            messages=messages,
+            max_tokens=max_tokens,
+            temperature=temperature,
+            top_p=top_p,
+            stream=True
+        )
+    
+        final_response = ""
+        for message in response.choices[0].message:
+            final_response += message.content
+    
+        return final_response
+    except OpenAIError as e:  
+        print(f"An error occurred while communicating with OpenAI: {e}")  
+        return None  
 
 demo = gr.ChatInterface(
     respond,