Update app.py

app.py

@@ -56,6 +56,11 @@ fig
 
 
 async def callback(content: str, user: str, instance: pn.chat.ChatInterface):
+    if not api_key_input.value:
+        yield "Please first enter your Mistral API key"
+        return
+    client = MistralAsyncClient(api_key=api_key_input.value)
+
     # system
     messages = [SYSTEM_MESSAGE]
 
@@ -87,7 +92,7 @@ def update_plot(event):
 
 
 # instantiate widgets and panes
-client = MistralAsyncClient(api_key=os.environ["MISTRAL_API_KEY"])
+api_key_input = pn.widgets.PasswordInput(placeholder="Enter your MistralAI API Key")
 chat_interface = pn.chat.ChatInterface(
     callback=callback,
     show_clear=False,
@@ -97,7 +102,7 @@ chat_interface = pn.chat.ChatInterface(
         show_reaction_icons=False,
         show_copy_icon=False,
     ),
-    height=700,
+    height=650,
     callback_exception="verbose",
 )
 matplotlib_pane = pn.pane.Matplotlib(
@@ -120,7 +125,7 @@ tabs = pn.Tabs(
     ("Code", code_editor),
 )
 
-sidebar = [chat_interface]
+sidebar = [api_key_input, chat_interface]
 main = [tabs]
 template = pn.template.FastListTemplate(
     sidebar=sidebar,