Upload app.py

app.py

@@ -3,6 +3,7 @@ import json
 import asyncio
 import os
 from EdgeGPT import Chatbot, ConversationStyle
+import re
 
 """
 参考资料：
@@ -25,13 +26,35 @@ async def get_model_reply(prompt,style,cookies,context=[]):
     raw_data = await bot.ask(prompt, conversation_style=style)
     await bot.close()
     #print(raw_data)
-    response = raw_data["item"]["messages"][1]["text"]
-    context += [response]
+    try:
+        try:
+            response = raw_data["item"]["messages"][1]["text"]
+        except:
+            response = raw_data["item"]["messages"][1]["adaptiveCards"][0]["body"][0]["text"]
+        response = re.sub(r'\^', '', response)
+        response = response.rstrip()
+        context += [response]
+        responses = [(u, b) for u, b in zip(context[::2], context[1::2])]
+        return responses, context
 
-    # list of (user, bot) responses. We will use this format later
-    responses = [(u, b) for u, b in zip(context[::2], context[1::2])]
+    except:
+        try:
+            if raw_data["item"]["throttling"]["numUserMessagesInConversation"] > raw_data["item"]["throttling"][
+                "maxNumUserMessagesInConversation"]:
+                response = ">>>请重新开启新的对话。"
+                print(response)
+                context += ["0"]
+                responses = [(u, b) for u, b in zip(context[::2], context[1::2])]
+                return responses, context
+
+        except:
+            if raw_data["item"]["result"]["value"] == "Throttled":
+                response = "> **错误： 我们很抱歉，但你已经达到了你在24小时内可以向Bing发送消息的最大数量。请稍后再查看！**"
+                print(response)
+                context += ["0"]
+                responses = [(u, b) for u, b in zip(context[::2], context[1::2])]
+                return responses, context
 
-    return responses, context
 
 with gr.Blocks() as dialog_app: