Update app.py

app.py

@@ -19,36 +19,48 @@ from g4f.Provider import (
 )
 import os
 
-from g4f.models import Model
-model_provider_dict = {}
+provider_dict = {
+    # 'Ails': Ails,
+    # 'You': You,
+    # 'Bing': Bing,
+    # 'Yqcloud': Yqcloud,
+    # 'Theb': Theb,
+    # 'Aichat': Aichat,
+    # 'Bard': Bard,
+    # 'Vercel': Vercel,
+    # 'Forefront': Forefront,
+    # 'Lockchat': Lockchat,
+    # 'Liaobots': Liaobots,
+    # 'H2o': H2o,
+    # 'ChatgptLogin': ChatgptLogin,
+    'DeepAi': DeepAi,
+    'GetGpt': GetGpt
+}
 
-# 遍历Model类中的name属性
-for name in vars(Model):
-    attr_value = getattr(Model, name)
-    if hasattr(attr_value, 'name'):
-        model_provider_dict[attr_value.name] = attr_value.best_provider
-
-# os.environ["no_proxy"] = "localhost,127.0.0.1,:1"
 
 with gr.Blocks() as demo:
-
   chatbot = gr.Chatbot([[None, None]], label='AI')
   msg = gr.Textbox(value="", label='')
   clear = gr.Button("Clear")
-  model_name = gr.Dropdown(model_provider_dict.keys())
+  with gr.Row():
+    model_name = gr.Dropdown(['gpt-3.5-turbo', 'gpt-4'], value='gpt-3.5-turbo', label='模型')
+    provider_name  =  gr.Dropdown(provider_dict.keys(), value='GetGpt', label='提供者')
 
   def user(user_message, history):
       return gr.update(value="", interactive=False), history + [[user_message, None]]
 
-  def bot(history, model_name):
+  def bot(history, model_name, provider_name):
       history[-1][1] = ''
-      bot_msg = g4f.ChatCompletion.create(model=model_name, provider=model_provider_dict[model_name], messages=[{"role": "user", "content": history[-1][0]}], stream=True)
+      bot_msg = g4f.ChatCompletion.create(model=model_name, 
+                                          provider=provider_dict[provider_name], 
+                                          messages=[{"role": "user", "content": history[-1][0]}],
+                                          stream=True)
       for c in bot_msg:
           history[-1][1] += c
           yield history
 
   response = msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-      bot, [chatbot, model_name], chatbot
+      bot, [chatbot, model_name, provider_name], chatbot
   )
   response.then(lambda: gr.update(interactive=True), None, [msg], queue=False)
   clear.click(lambda: None, None, chatbot, queue=False)