Update main.py

main.py

@@ -750,15 +750,15 @@ def handle_text_message(event):
         else:
             # chat = get_chat(event.source.user_id)
             # response = chat.send_message(text).text
-            response = query({"inputs": text,"parameters":{"return_full_text":False,"max_length":1024}})
+            response = query({"inputs": text,"parameters":{"return_full_text":False,"max_length":1024}})[0]["generated_text"]
 
-            app.logger.info("Got response with http status code: " + str(response))
-            # html_msg = markdown.markdown(response)
-            # soup = BeautifulSoup(html_msg, 'html.parser')
+            # app.logger.info("Got response with http status code: " + str(response))
+            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=str(response))]
+                    messages=[TextMessage(text=soup.get_text())]
                 )
             )