Update main.py

main.py

@@ -21,6 +21,7 @@ from langchain.schema.runnable import Runnable, RunnablePassthrough, RunnableCon
 from langchain.callbacks.base import BaseCallbackHandler
 
 import chainlit as cl
+from chainlit.input_widget import TextInput, Select, Switch, Slider
 from literalai import LiteralClient
 @cl.password_auth_callback
 def auth_callback(username: str, password: str):
@@ -93,6 +94,16 @@ model = HuggingFaceEndpoint(
 @cl.on_chat_start
 async def on_chat_start():
     await cl.Message(f"> REVIEWSTREAM").send()
+    settings = await cl.ChatSettings(
+        [
+            Select(
+                id="Model",
+                label="Publications de recherche",
+                values=["---", "HAL", "Persée"],
+                initial_index=0,
+            ),
+        ]
+    ).send()
     template = """Answer the question based only on the following context:
 
     {context}