Update app.py

app.py

@@ -489,15 +489,16 @@ def MoleculeAgent(smiles, name, task):
   for c in graph.stream(input): #, stream_mode='updates'):
     m = re.findall(r'[a-z]+\_node', str(c))
     if len(m) != 0:
-      reply = c[str(m[0])]['messages']
-      print("======== DEBUG ================")
-      print(c)
-      if 'assistant' in str(reply):
-        reply = str(reply).split("<|assistant|>")[-1].split('#')[0].strip()
-        replies.append(reply)
-    else:
-      reply = str(c).split("<|assistant|>")[-1].split('#')[0].strip()
-      replies.append(reply)
+      try:
+          reply = c[str(m[0])]['messages']
+          print("======== DEBUG ================")
+          print(c)
+          if 'assistant' in str(reply):
+            reply = str(reply).split("<|assistant|>")[-1].split('#')[0].strip()
+            replies.append(reply)
+      except:
+          reply = str(c).split("<|assistant|>")[-1].split('#')[0].strip()
+          replies.append(reply)
   #check if image exists
   if os.path.exists('Similars_image.png'):
     img_loc = 'Similars_image.png'