fixes

frontend.rb

@@ -28,16 +28,16 @@ file.write("Word: #{word}\n")
 
 Thread.new do
   cur_line = ''
-  is_thinking = false
+  thinks = nil
   while c = stdout.getc
     file.write(c)
     file.flush()
     if c == "\n"
       if cur_line.include?('<think>')
-        is_thinking = true
+        thinks = 0
       elsif cur_line.include?('</think>')
-        is_thinking = false
-        puts "</think>" # now we have 2 newlines
+        thinks = nil
+        puts "\n</think>" # now we have 2 newlines
       end
 
       cur_line = ''
@@ -45,14 +45,13 @@ Thread.new do
       cur_line += c
     end
 
-    if is_thinking
-      if c == ' '
-        print 'think '
-      elsif c == "\n"
-        puts
-      end
-    else
+    if thinks.nil?
       print c
+    else
+      if c == ' ' || c == "\n"
+        thinks += 1
+        print "\rthink x#{thinks}"
+      end
     end
   end
 end