Update app.py

app.py

@@ -20,10 +20,8 @@ from panel.chat import ChatInterface
 from panel.widgets import Button, CodeEditor, PasswordInput, Switch, TextInput
 
 pn.extension("codeeditor")
-
 template = pn.template.BootstrapTemplate(title=TITLE)
 
-
 def get_description_text():
     return f"""
     # {TITLE}
@@ -32,6 +30,30 @@ def get_description_text():
 
     #### [[AutoGen](https://github.com/microsoft/autogen)] [[Discord](https://discord.gg/pAbnFJrkgZ)] [[Paper](https://arxiv.org/abs/2308.08155)] [[SourceCode](https://github.com/thinkall/autogen-demos)]
     """
+    
+client = Client("https://tonic1-tulu.hf.space/--replicas/xh5ff/")  
+  
+async def send_messages_with_tulua_pi(recipient,messages,sender):  
+      response_content =""  
+        
+      if len(messages) >0:  
+          message_text= messages[-1]['content']  
+          result await client.predict(message_text)  
+            
+          if 'message' in result :  
+              reponse_content=result['message']  
+                           
+      chatiface.send(reponse_content,user=sender.name,resend=False)  
+
+# Define a panel Row for OpenAI settings.  
+openai_row = pn.Row(txt_model, pwd_openai_key, pwd_openai_url)  
+  
+def check_yi_mode(event):  
+    # Make openai_row invisible when local YI Mode is enabled and vice versa.  
+    openai_row.visible = not event.new  
+  
+# Add an observer to watch changes in yi_mode value (True/False)  
+YI_MODE.param.watch(check_yi_mode, 'value')  
 
 
 template.main.append(
@@ -65,12 +87,17 @@ pwd_aoai_url = PasswordInput(
     sizing_mode="stretch_width",
 )
 file_cfg = pn.widgets.FileInput(filename="OAI_CONFIG_LIST", sizing_mode="stretch_width")
-template.main.append(
-    pn.Row(
-        txt_model, pwd_openai_key, pwd_openai_url, pwd_aoai_key, pwd_aoai_url, file_cfg
+template.main.append(  
+  pn.Column(  
+      openai_row,  
+      [pn.pane.Markdown("### Local YI Mode: "),Yi_Mode],
+      [pn.pane.Markdown("### Yi Endpoint: "), yi_endpoint]  
     )
 )
 
+# Add YI Mode Checkbox and Endpoint Input.  
+YI_MODE = pn.widgets.Checkbox(name='Local Y.I. Mode', value=True)   
+YI_ENDPOINT = pn.widgets.TextInput(name='Yi Endpoint', placeholder="Enter your YI endpoint here...")  
 
 def get_config(tmpfilename="OAI_CONFIG_LIST"):
     os.makedirs(".chromadb", exist_ok=True)
@@ -85,6 +112,14 @@ def get_config(tmpfilename="OAI_CONFIG_LIST"):
         cfg_fpath,
         file_location=".",
     )
+    if YI_MODE == "local": 
+        config_list = [
+        {
+            "model": "EasyYI",  
+            "api_key": "None",
+            "base_url": "https://huggingface.co/spaces/teamtonic/easyyi"
+        }
+    ] 
     if not config_list:
         os.environ["MODEL"] = txt_model.value
         os.environ["OPENAI_API_KEY"] = pwd_openai_key.value