update

app.py

@@ -21,19 +21,18 @@ def completion(prompt):
         'stop': None,
         # 'stop': [[13, 13]],
     }, headers={"Authorization": f"Bearer {st.secrets['token']}"})
-
-    st.success(str(resp))
-    st.success(resp.text)
-    print(resp, '\n', resp.text)
-    answers = resp.json()
-    answers = [c['text'] for c in answers['choices'] if c['text'] is not None]
-    cols = st.columns(3)
-    for idx, answer in enumerate(answers):
-        if idx >= 3:
-            break
-        with cols[idx]:
-            content = (prompt + answer).replace("\n", "\n\n")
-            st.markdown(f'## 版本{idx}\n\n{content}')
+    if resp.status_code != 200:
+        st.error(f'Bad response: {resp}')
+    else:
+        answers = resp.json()
+        answers = [c['text'] for c in answers['choices'] if c['text'] is not None]
+        cols = st.columns(3)
+        for idx, answer in enumerate(answers):
+            if idx >= 3:
+                break
+            with cols[idx]:
+                content = (prompt + answer).replace("\n", "\n\n")
+                st.markdown(f'## 版本{idx}\n\n{content}')
     end = time.monotonic()
     st.text(f'耗时：{end - start}')