Update app.py

app.py

@@ -152,11 +152,16 @@ def merlin_chat(message, history):
         #start = cleanStr.find('{')
         #if start<=0:
         #    continue
-        end = cleanStr.find('\"')
+        end = cleanStr.find('}') + 1
         if end<=0:
             continue
         cleanStr = cleanStr[:end]
-        return message['cleanStr']
+        messageStr = prompt + cleanStr + ']'
+        messages = json.loads(messageStr)
+        message = messages[-1]
+        if message['role'] != 'assistant':
+            continue
+        return message['content']
     return "🤔"
 
 gr.ChatInterface(merlin_chat).launch()