Update app.py

app.py

@@ -320,20 +320,27 @@ does not inform about features unique to each molecule.'
 
   tool_choices = str(res).split('<|assistant|>')[1].split('#')[0].strip()
   tool_choices = tool_choices.split(',')
+
   if len(tool_choices) == 1:
-    if tool_choices[0].strip().lower() == 'none':
+    tool1 = tool_choices[0].strip()
+    if tool1.lower() == 'none':
       tool_choice = (None, None)
     else:
-      tool_choice = (tool_choices[0].strip().lower(), None)
+      tool_choice = (tool1, None)
   elif len(tool_choices) == 2:
-    if tool_choices[0].strip().lower() == 'none':
-      tool_choice = (None, tool_choices[1].strip().lower())
-    elif tool_choices[1].strip().lower() == 'none':
-      tool_choice = (tool_choices[0].strip().lower(), None)
+    tool1 = tool_choices[0].lower().strip()
+    tool2 = tool_choices[1].lower().strip()
+    if tool1.lower() == 'none' and tool2.lower() == 'none':
+      tool_choice = (None, None)
+    elif tool1.lower() == 'none' and tool2.lower() != 'none':
+      tool_choice = (None, tool2)
+    elif tool2.lower() == 'none' and tool1.lower() != 'none':
+      tool_choice = (tool1, None)
     else:
-      tool_choice = (tool_choices[0].strip().lower(), tool_choices[1].strip().lower())
+      tool_choice = (tool1, tool2)
   else:
     tool_choice = (None, None)
+    
   state["tool_choice"] = tool_choice
   state["which_tool"] = 0
   print(f"The chosen tools are: {tool_choice}")