Update app.py

app.py

@@ -25,6 +25,7 @@ from g4f.Provider import (
     BingHuan,
     Wewordle,
     ChatgptAi,
+    opchatgpts
 )
 import os
 import json
@@ -47,7 +48,6 @@ from langchain.utilities import WikipediaAPIWrapper
 from langchain.tools import DuckDuckGoSearchRun
 from models_for_langchain.memory_func import validate_memory_len
 
-
 provider_dict = {
     'Ails': Ails,
     'You': You,
@@ -72,15 +72,13 @@ provider_dict = {
     'BingHuan': BingHuan,
     'Wewordle': Wewordle,
     'ChatgptAi': ChatgptAi,
+    'opchatgpts': opchatgpts
 }
 
-available_dict = {
-    'gpt-3.5-turbo':['Acytoo', 'AiService', 'Aichat', 'GetGpt', 'Wewordle'],
-    'gpt-4':['ChatgptAi'],
-    'falcon-7b':['H2o'],
-    'falcon-13b':['H2o'],
-    'llama-13b':['H2o']
-}
+
+with open("available_dict.txt", "r") as fp:
+    # Load the dictionary from the file
+    available_dict = json.load(fp)
 
 def change_prompt_set(prompt_set_name):
     return gr.Dropdown.update(choices=list(prompt_set_list[prompt_set_name].keys()))
@@ -141,8 +139,13 @@ def bot(history, model_name, provider_name, system_msg, agent):
         prompt = new_template.format(
             chat_history = prev_memory,
         )
-        # print(f'prompt = \n --------\n{prompt}\n --------')
-        bot_msg = llm._call(prompt=prompt)
+        print(f'prompt = \n --------\n{prompt}\n --------')
+        for _ in range(3):
+            try:
+                bot_msg = llm._call(prompt=prompt)
+                break
+            except:
+                bot_msg = '服务器无响应，请更换提供者或者清空对话后重试。'
 
         for c in bot_msg:
             history[-1][1] += c
@@ -187,8 +190,8 @@ with gr.Blocks() as demo:
     memory = ConversationBufferWindowMemory(k=6, memory_key="chat_history")
     with gr.Row():
         model_name = gr.Dropdown(list(available_dict.keys()), value='gpt-3.5-turbo', label='模型')
-        provider = gr.Dropdown(available_dict['gpt-3.5-turbo'], value='AiService', label='提供者', min_width=20)
-        agent = gr.Dropdown(['系统提示', '维基百科', 'duckduckgo'], value='系统提示', label='Agent')
+        provider = gr.Dropdown(available_dict['gpt-3.5-turbo'], value='GetGpt', label='提供者', min_width=20)
+        agent = gr.Dropdown(['系统提示', '维基百科'], value='系统提示', label='Agent')
     system_msg = gr.Textbox(value="你是一名助手，可以解答问题。", label='系统提示')
     
     chatbot = gr.Chatbot([[None, None]], label='AI')