Update main.py

main.py

@@ -748,15 +748,15 @@ def handle_text_message(event):
                 app.logger.info("Got x-line-request-id: " + e.headers['x-line-request-id'])
                 app.logger.info("Got response with http body: " + str(ErrorResponse.from_json(e.body)))
         else:
-            chat = get_chat(event.source.user_id)
-            response = chat.send_message(text).text
-            # response = query({"inputs": input,"parameters":{"return_full_text":False,"max_length":1024}})
+            # chat = get_chat(event.source.user_id)
+            # response = chat.send_message(text).text
+            response = query({"inputs": input,"parameters":{"return_full_text":False,"max_length":1024}})[0]["generated_text"]
             html_msg = markdown.markdown(response)
             soup = BeautifulSoup(html_msg, 'html.parser')
             line_bot_api.reply_message(
                 ReplyMessageRequest(
                     reply_token=event.reply_token,
-                    messages=[TextMessage(text=response)]
+                    messages=[TextMessage(text=soup.get_text())]
                 )
             )