LocalGPT / chainlit_app.py
Taoheed-O's picture
new space
d0f4923
# import chainlit as cl
# @cl.on_message
# def main(message:str):
# result = message.title()
# cl.Message(content= f"sure here is your message: {result}").send()
# @cl.on_chat_start
# def chat():
# content = "Hi welcome to teslabot what can I do for you?"
# cl.Message(content=content).send()
import chainlit as cl
from textblob import TextBlob
from gpt4all import GPT4All
gpt = GPT4All(model_name='nous-hermes-13b.ggmlv3.q4_0.bin', model_path='models')
@cl.on_chat_start
async def start():
# Send the first message without the elements
content = "Hi, this is TeslaBot how can I help you?"
await cl.Message(
content=content,
).send()
# elements = [
# cl.Image(path="images/avatar.jpeg", name="image1", display="inline"),
# cl.Text(content="Here is a side text document", name="text1", display="side"),
# cl.Text(content="Here is a page text document", name="text2", display="page"),
#]
# Send the second message with the elements
# await cl.Message(
# content=content,
# #elements=elements,
# ).send()
@cl.on_message
async def message(text:str):
# logic
if 'count' in text:
file = None
while file == None:
file = await cl.AskFileMessage(
content='please upload the file here',
accept=['text/plain']).send()
text = file[0].content.decode("utf-8")
count = len(text)
await cl.Message(content = f'The word count of your text is: {count} ').send()
else:
response = gpt.generate(text, temp=0.65)
await cl.Message(f'{response}')