Update app.py

app.py

@@ -86,14 +86,11 @@ async def start_chat():
 async def main(message: cl.Message):
     if "document:" in message.content.lower() or "documents:" in message.content.lower():
         # Prepare the message with documents context
-        prompt = RAG_Trwira.main(message.content[10:]) #slice the "documents" command
+        prompt = RAG_Trwira.main(message.content[10:]) # slice the "documents" command
     else:
         # Without documents context
         prompt = message.content
     
-    # Sanity check
-    # print(prompt)
-    
     # Format the messages as a list of message dictionaries
     message_formated = [
         {"role": "user", "content": prompt}
@@ -111,4 +108,4 @@ async def main(message: cl.Message):
             await msg.stream_token(token)
 
     # Update the message after streaming completion
-    await msg.update()
+    await msg.update()